vladyur commited on
Commit
a16dba0
·
1 Parent(s): ded5ed0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def get_model(model_name, model_path):
14
  return model, tokenizer
15
 
16
 
17
- @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
18
  def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_length=300):
19
  input_ids = tokenizer.encode(text, return_tensors="pt")
20
  with torch.no_grad():
 
14
  return model, tokenizer
15
 
16
 
17
+ #@st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, tokenizers.AddedToken: lambda _: None, re.Pattern: lambda _: None}, allow_output_mutation=True, suppress_st_warning=True)
18
  def predict(text, model, tokenizer, n_beams=5, temperature=2.5, top_p=0.8, max_length=300):
19
  input_ids = tokenizer.encode(text, return_tensors="pt")
20
  with torch.no_grad():