Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,15 @@
 import streamlit as st
 from transformers import pipeline
 
-# ---------------------------------------------------
-# PAGE CONFIG
-# ---------------------------------------------------
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
 # ---------------------------------------------------
-# LOAD MODEL
+# LOAD MODEL (FIXED FOR FLAN-T5)
 # ---------------------------------------------------
 @st.cache_resource
 def load_model():
     return pipeline(
-        "text-generation",
+        "text2text-generation",   # ✅ FIXED
         model="google/flan-t5-base"
     )
 
@@ -166,7 +163,7 @@ if st.button("🚀 Submit Profile"):
         bmi_status = bmi_category(bmi)
         equipment_list = ", ".join(equipment)
 
-       # Improved Prompt
+        # Improved Prompt
         prompt = f"""
 You are a certified professional fitness trainer.
 
@@ -188,15 +185,11 @@ Day 1:
             response = generator(
                 prompt,
                 max_new_tokens=400,
-                do_sample=True,
-                temperature=0.7
+                temperature=0.7,
+                do_sample=True
             )
 
-            result = response[0]["generated_text"]
-
-            # Remove prompt if model repeats it
-            if result.startswith(prompt):
-                result = result[len(prompt):].strip()
+            result = response[0]["generated_text"].strip()
 
         st.subheader("🏋️ Your Personalized Workout Plan")
         st.write(result)