Update src/streamlit_app.py

src/streamlit_app.py

@@ -12,7 +12,7 @@ st.set_page_config(page_title="FitPlan AI", layout="centered")
 @st.cache_resource
 def load_model():
     return pipeline(
-        "text2text-generation",
+        "text-generation",
         model="google/flan-t5-base"
     )
 
@@ -186,7 +186,13 @@ if st.button("🚀 Submit Profile"):
         """
 
         with st.spinner("Generating your AI workout plan..."):
-            response = generator(prompt,max_length=512)
+            response = generator(
+                prompt,
+                max_new_tokens=400,
+                do_sample=True,
+                temperature=0.7
+            )
+
             result = response[0]["generated_text"]
 
         st.subheader("🏋️ Your Personalized Workout Plan")