Spaces:
Build error
Build error
extend the timeout
Browse files- app/app.py +1 -1
app/app.py
CHANGED
@@ -66,7 +66,7 @@ def get_generator(model_name: str):
|
|
66 |
# Disable the st.cache for this function due to issue on newer version of streamlit
|
67 |
# @st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: id})
|
68 |
def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
|
69 |
-
temperature: float = 1.0, max_time: float =
|
70 |
# st.write("Cache miss: process")
|
71 |
set_seed(seed)
|
72 |
result = text_generator(text, max_length=max_length, do_sample=do_sample,
|
|
|
66 |
# Disable the st.cache for this function due to issue on newer version of streamlit
|
67 |
# @st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: id})
|
68 |
def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
|
69 |
+
temperature: float = 1.0, max_time: float = 120.0, seed=42):
|
70 |
# st.write("Cache miss: process")
|
71 |
set_seed(seed)
|
72 |
result = text_generator(text, max_length=max_length, do_sample=do_sample,
|