Update src/streamlit_app.py

src/streamlit_app.py

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