Update app.py

app.py

@@ -1,5 +1,6 @@
 import random
 import torch
+import numpy as np
 from tqdm import tqdm
 from functools import partialmethod
 import gradio as gr
@@ -41,7 +42,7 @@ def dalle_wrapper(prompt: str):
         top_p=top_p
     )
     title = f"<b>{prompt}</b>"
-    return title, images[0]
+    return title, np.array(images[0])
 
 
 translator = gr.Interface(fn=translation_wrapper,