Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,8 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
+import random
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
 
 # -------------------------
 # Page Config
@@ -111,8 +113,8 @@ if st.button("Generate Workout Plan"):
 
             outputs = model.generate(
                 **inputs,
-                max_new_tokens=700,
-                temperature=0.3,
+                max_new_tokens=400,
+                temperature=0.9,
                 do_sample=True
             )