Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,7 +6,9 @@ from transformers import pipeline
 # ---------------------------------------------------
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-
+# ---------------------------------------------------
+# LOAD MODEL
+# ---------------------------------------------------
 @st.cache_resource
 def load_model():
     return pipeline(
@@ -57,6 +59,9 @@ with col2:
         step=0.1
     )
 
+# ---------------------------------------------------
+# BMI FUNCTION
+# ---------------------------------------------------
 def bmi_category(bmi):
     if bmi < 18.5:
         return "Underweight"
@@ -67,7 +72,9 @@ def bmi_category(bmi):
     else:
         return "Obese"
 
-
+# ---------------------------------------------------
+# BMI CALCULATION
+# ---------------------------------------------------
 bmi = None
 
 if height > 0 and weight > 0:
@@ -179,7 +186,8 @@ if st.button("🚀 Submit Profile"):
         """
 
         with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
+            response = generator(prompt, max_new_tokens=400)
+            result = response[0]["generated_text"]
 
         st.subheader("🏋️ Your Personalized Workout Plan")
         st.write(result)