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(
-        "text-generation",
+        "text2text-generation",
         model="google/flan-t5-base"
     )
 
@@ -186,7 +186,7 @@ if st.button("🚀 Submit Profile"):
         """
 
         with st.spinner("Generating your AI workout plan..."):
-            response = generator(prompt, max_new_tokens=400)
+            response = generator(prompt,max_length=512)
             result = response[0]["generated_text"]
 
         st.subheader("🏋️ Your Personalized Workout Plan")