Update src/streamlit_app.py

src/streamlit_app.py

@@ -29,11 +29,10 @@ def get_category(bmi):
 @st.cache_resource
 def load_model():
     return pipeline(
-        task="text2text-generation",
+        task="image-text-to-text",
         model="google/flan-t5-base",
         tokenizer="google/flan-t5-base"
     )
-
 generator = load_model()
 
 # -------------------------