Update diffusion.py

diffusion.py

@@ -1,5 +1,13 @@
 import gradio as gr
+from PIL import Image
+import io 
 
+def im_2_bytes(img_path):
+    pil_im=Image.open(img_path)
+    b = io.BytesIO()
+    pil_im.save(b, 'png')
+    im = b.getvalue()
+    return (im)
 def gen_image(prompt,model_name="dreamlike-art/dreamlike-photoreal-2.0",count = 1):
     proc=gr.load(f"models/{model_name}")
     t=0
@@ -9,4 +17,6 @@ def gen_image(prompt,model_name="dreamlike-art/dreamlike-photoreal-2.0",count =
         output_list.append(output)
         inputs=f"{inputs} "
         t+=1
-    return(output_list)
+    #return(output_list)
+    return(im_2_bytes(output))
+