Update src/streamlit_app.py

src/streamlit_app.py

@@ -18,8 +18,8 @@ st.write("Fill your details to generate AI workout plan.")
 
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
     return tokenizer, model
 
 tokenizer, model = load_model()
@@ -111,8 +111,8 @@ if st.button("Generate Workout Plan"):
 
             outputs = model.generate(
                 **inputs,
-                max_new_tokens=300,
-                temperature=0.7,
+                max_new_tokens=700,
+                temperature=0.3,
                 do_sample=True
             )