Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,10 +5,9 @@ import torch
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
 # LOAD MODEL 
-@st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
     return tokenizer, model
 
 tokenizer, model = load_model()