Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,10 +1,16 @@
 import streamlit as st
 from transformers import pipeline
-import streamlit as st
 
 st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
 
 st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
+def load_model():
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
+ 
+generator = load_model()
 
 st.write("Fill in your details to calculate your BMI and fitness category.")