Update src/streamlit_app.py

src/streamlit_app.py

@@ -26,12 +26,11 @@ def get_category(bmi):
 # -------------------------
 @st.cache_resource
 def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
-    )
-
-generator = load_model()
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    return tokenizer, model
+ 
+tokenizer, model = load_model()
 
 # -------------------------
 # UI STARTS