Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,38 +1,38 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load model
+# Cache model to avoid reloading
 @st.cache_resource
 def load_model():
     return pipeline(
         "text2text-generation",
-        model="google/flan-t5-base"
+        model="google/flan-t5-small"   # smaller & safer for HF Spaces
     )
 
-model = load_model()
+generator = load_model()
 
 st.set_page_config(page_title="Fitness Plan AI", layout="centered")
 
 st.title("🏋️ Fitness Plan AI Generator")
 st.write("Generate a personalized fitness plan using AI")
 
-# User inputs
-age = st.number_input("Age", 15, 80, 22)
+# User Inputs
+age = st.number_input("Age", min_value=15, max_value=80, value=22)
 gender = st.selectbox("Gender", ["Male", "Female"])
 goal = st.selectbox("Fitness Goal", ["Weight Loss", "Muscle Gain", "General Fitness"])
 level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
 days = st.slider("Workout Days per Week", 1, 7, 4)
 
 if st.button("Generate Fitness Plan"):
-    prompt = f"""
-    Create a {days}-day fitness plan for a {age}-year-old {gender}.
-    Goal: {goal}
-    Level: {level}
-    Include warm-up, main workout, and cool-down.
-    """
-
-    with st.spinner("Generating plan..."):
-        output = model(prompt, max_length=300)
-        st.success("Your Fitness Plan:")
-        st.write(output[0]['generated_text'])
+    prompt = (
+        f"Create a {days}-day fitness workout plan for a "
+        f"{age}-year-old {gender}. "
+        f"Goal: {goal}. "
+        f"Fitness level: {level}. "
+        f"Include warm-up, main workout, and cool-down."
+    )
 
+    with st.spinner("Generating your fitness plan..."):
+        result = generator(prompt, max_length=300, do_sample=True)
+        st.success("✅ Your Fitness Plan")
+        st.write(result[0]["generated_text"])