cahya commited on
Commit
c5ea533
1 Parent(s): fcc6e50

extend the max_time to 20 secs

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