Update app.py

app.py

@@ -14,9 +14,9 @@ def predict(text_input):
 iface = gr.Interface(
     fn=predict,
     inputs=gr.Textbox(lines=5, label="Input Text"),
-    outputs="text",
+    outputs=gr.Image(type="numpy", label="Output Image"),
     title="Diffusion Pipeline App",
-    description="Enter the text to get the model's predictions on the input."
+    description="Enter the text to get the model's predictions as an output image."
 )
 
 # Launch the Gradio interface