cahya commited on
Commit
2744cb8
1 Parent(s): 22576bc

extend the timeout

Browse files
Files changed (1) hide show
  1. 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 = 60.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,
 
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,