srustik123 commited on
Commit
a62fdd7
·
verified ·
1 Parent(s): 4bbe492

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -29,7 +29,7 @@ def get_category(bmi):
29
  def load_model():
30
  return pipeline(
31
  "text-generation",
32
- model="gpt2"
33
  )
34
 
35
  generator = load_model()
 
29
  def load_model():
30
  return pipeline(
31
  "text-generation",
32
+ model="google/flan-t5-base"
33
  )
34
 
35
  generator = load_model()