Update src/streamlit_app.py

src/streamlit_app.py

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