Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,9 +3,11 @@ from transformers import pipeline
 # 1. Basic Configuration
 st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
 def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    return tokenizer, model
+ 
+tokenizer, model = load_model()
     )
  
 generator = load_model()