Hugging Face's logo

Spaces:
Modfiededition
/
Writing_Assistant

App Files Community
Modfiededition commited on
Commit
6b29a33
•
1 Parent(s): acccb23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model_checkpoint = "huggingface/models/t5-base-fine-tuned-on-jfleg"
10
 
11
  @st.cache
12
  def load_model(model_name):
13
- translator= pipeline("text2text: generation", model=model_checkpoint)
14
  return translator
15
 
16
  translator= load_model(model_checkpoint)
 
10
 
11
  @st.cache
12
  def load_model(model_name):
13
+ translator= pipeline("text2text-generation", model=model_checkpoint)
14
  return translator
15
 
16
  translator= load_model(model_checkpoint)