Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,8 +7,8 @@ st.set_page_config(page_title="FitPlan AI", layout="centered")
 # LOAD MODEL 
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
     return tokenizer, model
 
 tokenizer, model = load_model()
@@ -170,7 +170,7 @@ Day 1:
 
             outputs = model.generate(
                 **inputs,
-                max_new_tokens=300,
+                max_new_tokens=900,
                 temperature=0.7,
                 do_sample=True
             )