Update app.py

app.py

@@ -31,13 +31,14 @@ def gen_image(desc: str, style: str):
     # image = Image.fromarray(np.uint8(img_np))
     img_rep = requests.get(resp_data['output_image_url'][0])
     image = Image.open(BytesIO(img_rep.content))
+    image_np = np.asarray(image)
 
-    return [image]
+    return image_np
 
 
 demo = gr.Interface(
     fn=gen_image,
     inputs=["text", "text"],
-    outputs=["image"],
+    outputs=gr.outputs.Image(type='numpy'),
 )
 demo.launch()