Update app.py

app.py

@@ -228,8 +228,8 @@ def generate_image(prompt):
 
 iface = gr.Interface(
     fn=generate_image,
-    inputs=gr.inputs.Textbox(label="Enter prompt", type="bytes"),
-    outputs="text",
+    inputs=gr.inputs.Textbox(label="Enter prompt"),
+    outputs=gr.outputs.Image(label="Generated Image"),
     title="KVIImager 2.0 - Image Generator",
     description="Generate images based on a prompt.",
     examples=[