print output in gradio box

app.py

@@ -28,9 +28,9 @@ def text_to_image(image, prompt):
     output = model.generate(**inputs, max_new_tokens=100)
     generated_text = processor.batch_decode(output, skip_special_tokens=True)
     for text in generated_text:
-        print(text.split("ASSISTANT:")[-1])
+        return text.split("ASSISTANT:")[-1]
+
 
-print()
 demo = gr.Interface(
     fn=text_to_image,
     inputs=[