Update src/streamlit_app.py

src/streamlit_app.py

@@ -28,7 +28,10 @@ def get_category(bmi):
 # -------------------------
 @st.cache_resource
 def load_model():
-    return pipeline("text2text-generation", model="google/flan-t5-base")
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
 
 generator = load_model()