Update app.py

app.py

@@ -108,7 +108,13 @@ def generate_image(prompt, image):
 
 
 if __name__ == "__main__":
-    demo = gr.Interface(fn=generate_image, inputs=["text", "image"], outputs=["image"])
+        demo = gr.Interface(fn=generate_image, inputs=[
+            "text",
+            gr.Image(image_mode='RGBA', type="numpy")
+        ],
+        outputs=[
+            gr.Image(type="numpy", image_mode='RGBA')
+        ])
     demo.launch(debug=True)
     logger.debug('demo.launch()')