Update src/streamlit_app.py

src/streamlit_app.py

@@ -7,8 +7,8 @@ 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-base")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
     return tokenizer, model
 
 tokenizer, model = load_model()
@@ -182,10 +182,12 @@ Equipment: {equipment_list}
             inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
 
             outputs = model.generate(
-                **inputs,
-                max_new_tokens=300,
-                temperature=0.7,
-                do_sample=True
+                inputs,
+                max_new_tokens=600,
+                temperature=0.3,
+                do_sample=False,
+                top_p=0.9,
+                repetition_penalty=1.2
             )
 
             result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()