Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,12 +8,11 @@ st.set_page_config(page_title="FitPlan AI", layout="centered")
 # ---------------------------------------------------
 @st.cache_resource
 def load_model():
-    return pipeline(
-        "text2text-generation",   # ✅ FIXED
-        model="google/flan-t5-base"
-    )
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    return tokenizer, model
 
-generator = load_model()
+tokenizer, model = load_model()
 
 # ---------------------------------------------------
 # TITLE
@@ -182,14 +181,14 @@ Day 1:
 """
 
         with st.spinner("Generating your AI workout plan..."):
-            response = generator(
-                prompt,
-                max_new_tokens=400,
-                temperature=0.7,
-                do_sample=True
-            )
+            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+
+outputs = model.generate(
+    **inputs,
+    max_new_tokens=400,
+    temperature=0.7,
+    do_sample=True
+)
 
-            result = response[0]["generated_text"].strip()
+result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
 
-        st.subheader("🏋️ Your Personalized Workout Plan")
-        st.write(result)