ZeeAI1 commited on
Commit
cc7449e
·
verified ·
1 Parent(s): d2bc406

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import time
7
  # Load models
8
  @st.cache_resource
9
  def load_grammar_model():
10
- return pipeline("text2text-generation", model="pszemraj/flan-t5-grammar-correction")
11
 
12
  @st.cache_resource
13
  def load_explainer_model():
 
7
  # Load models
8
  @st.cache_resource
9
  def load_grammar_model():
10
+ return pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
11
 
12
  @st.cache_resource
13
  def load_explainer_model():