dla9944 commited on
Commit
1157b5f
·
1 Parent(s): d64131c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from transformers import pipeline
8
 
9
  model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg"
10
 
11
-
12
- @st.cache(allow_output_mutation=True, suppress_st_warning=True)
13
  def load_model():
14
  return pipeline("text2text-generation", model=model_checkpoint)
15
  model = load_model()
 
8
 
9
  model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg"
10
 
11
+ @st.cache_data(allow_output_mutation=True)
12
+ # @st.cache_(allow_output_mutation=True, suppress_st_warning=True)
13
  def load_model():
14
  return pipeline("text2text-generation", model=model_checkpoint)
15
  model = load_model()