Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,12 +1,16 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
 
-# Load model only once
+# Load model (Stable version)
 @st.cache_resource
 def load_model():
-    return pipeline("text-generation", model="google/flan-t5-base")
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
 
-generator = load_model()
+tokenizer, model = load_model()    
+
+# Page Config
 
 st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
 st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
@@ -103,7 +107,14 @@ if st.button("Submit Profile"):
         """
 
         with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
+            inputs = tokenizer(prompt, return_tensors="pt", truncation= True)
+            outputs = model.generate(
+                **inputs,
+                max_new_tokens=600,
+                temperature= 0.7
+                do_sample= True
+            )
+            result = tokenizer.decode(outputs[0], skip_special_tokens= True)
 
         st.subheader("🏋️ Your Personalized Workout Plan")
         st.write(result)