ljyflores commited on
Commit
68b6abc
·
1 Parent(s): 851657f

Fix cache_data issue

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -34,8 +34,8 @@ def load(dataset_name, model_variant_name):
34
  )
35
 
36
  @st.cache_data()
37
- def predict(text, pipeline):
38
- return pipeline(text, max_length=768, do_sample=False)
39
 
40
  def clean(s):
41
  return s.replace("<s>","").replace("</s>","")
 
34
  )
35
 
36
  @st.cache_data()
37
+ def predict(text, _pipeline):
38
+ return _pipeline(text, max_length=768, do_sample=False)
39
 
40
  def clean(s):
41
  return s.replace("<s>","").replace("</s>","")