Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from transformers import pipeline
 
 def calculate_bmi(weight, height_cm):
     # Convert height from cm to meters
@@ -20,6 +21,13 @@ def get_category(bmi):
 # App UI
 st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
 st.title("💪 FitPlan-AI: Personalized Fitness Profile")
+def load_model():
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
+ 
+generator = load_model()
 
 with st.form("fitness_form"):
     st.header("Personal Information")