Update src/streamlit_app.py

src/streamlit_app.py

@@ -13,18 +13,12 @@ st.set_page_config(
 # =========================
 # Load Model
 # =========================
-@st.cache_resource
 def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
-    )
-
-generator = load_model()
-
-st.title("🏋️ Personalized Fitness Profile")
-st.markdown("---")
-
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    return tokenizer, model
+ 
+tokenizer, model = load_model()
 # =========================
 # 1️⃣ Personal Information
 # =========================