Spaces:
Runtime error
Runtime error
extend max_time to 60
Browse files- app/app.py +1 -1
app/app.py
CHANGED
@@ -37,7 +37,7 @@ def get_generator(model_name: str):
|
|
37 |
|
38 |
@st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: id})
|
39 |
def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
|
40 |
-
temperature: float = 1.0, max_time: float =
|
41 |
# st.write("Cache miss: process")
|
42 |
set_seed(seed)
|
43 |
result = text_generator(text, max_length=max_length, do_sample=do_sample,
|
|
|
37 |
|
38 |
@st.cache(suppress_st_warning=True, hash_funcs={tokenizers.Tokenizer: id})
|
39 |
def process(text_generator, text: str, max_length: int = 100, do_sample: bool = True, top_k: int = 50, top_p: float = 0.95,
|
40 |
+
temperature: float = 1.0, max_time: float = 60.0, seed=42):
|
41 |
# st.write("Cache miss: process")
|
42 |
set_seed(seed)
|
43 |
result = text_generator(text, max_length=max_length, do_sample=do_sample,
|