Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,13 +2,16 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 
-# Load model (Stable version)    
 @st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
     model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
     return tokenizer, model
 
+# ⭐ MUST CALL FUNCTION
+tokenizer, model = load_model()
+
+
 # Page Config
 
 st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
@@ -40,7 +43,7 @@ goal = st.selectbox(
 
 equipment = st.multiselect(
     "Available Equipment",
-    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope", "Weight plates", "Cycling", "Inclined bench","Pullups Bar", "No Equipment"]
 )
 
 fitness_level = st.radio(