streamlit write

app.py

@@ -26,4 +26,4 @@ prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_genera
 if st.button("generate response"):
 
     outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
-    print(outputs[0]["generated_text"])
+    st.write(outputs[0]["generated_text"])