Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import transformers # Changed the way we import to check for errors
+import transformers 
 from transformers import pipeline 
 
 # --- Page Config ---
@@ -23,7 +23,7 @@ if 'user_data' not in st.session_state:
 if 'profile_complete' not in st.session_state:
    st.session_state.profile_complete = False
 
-# --- Helper Functions ---
+# ---  Functions ---
 def calculate_bmi(w, h_cm):
    if h_cm > 0:
        return round(w / ((h_cm / 100) ** 2), 2)
@@ -36,7 +36,7 @@ def bmi_category(bmi):
    if bmi < 30: return "Overweight"
    return "Obese"
 
-# --- Mam's Code: Model Loading ---
+# - ---
 @st.cache_resource 
 def load_model():
     # Using text-generation as per the environment's supported list
@@ -66,7 +66,7 @@ if menu == "👤 Profile":
 
    bmi = calculate_bmi(u_weight, u_height)
 
-   # --- Mam's Code: Submit Button & Validation ---
+   #  ---
    if st.button("Submit Profile"):
     
     if not name:
@@ -87,7 +87,7 @@ if menu == "👤 Profile":
         bmi_status = bmi_category(bmi)
         equipment_list = ", ".join(equipment)
  
-        # Mam's Code: Prompt
+        # Prompt
         prompt = f"""
         Generate a 5-day structured workout plan.
  
@@ -108,7 +108,7 @@ if menu == "👤 Profile":
         """
  
         with st.spinner("Generating your AI workout plan..."):
-            # Mam's Code: Generator call
+            
             result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
  
         st.subheader("🏋️ Your Personalized Workout Plan")