Update src/streamlit_app.py

src/streamlit_app.py

@@ -180,17 +180,18 @@ Equipment: {equipment_list}
         with st.spinner("Generating your AI workout plan..."):
 
             inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
-
+            
             outputs = model.generate(
-                inputs,
+                **inputs,
                 max_new_tokens=600,
                 temperature=0.3,
                 do_sample=False,
-                top_p=0.9,
                 repetition_penalty=1.2
             )
+            
+            response = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-            result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+st.write(response)
 
         st.subheader("🏋️ Your Personalized Workout Plan")
         st.write(result)