Update app.py

app.py

@@ -1,6 +1,9 @@
 import os
 import gradio as gr
 import openai
+from PIL import Image
+import io
+import numpy as np
 
 openai.api_key = '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' # your api key
 openai.Model.list()
@@ -15,7 +18,12 @@ def DALLE(user_input):
         size="1024x1024"
     )
     
-    return DALLE_reply
+    # Convert the OpenAI output to an image format that Gradio can recognize
+    img_bytes = io.BytesIO(DALLE_reply.get()['data'])
+    img = Image.open(img_bytes)
+    img_arr = np.array(img)
+    
+    return img_arr
 
 iface = gr.Interface(fn=DALLE, inputs="text", outputs="image", title="bhAI")
-iface.launch()
+iface.launch()