resize back 32

app.py

@@ -76,9 +76,9 @@ def process_images(image,prompt="a girl",strength=0.75,seed=0,inference_step=4,p
         # resize back multiple of 32
         if (new_width!=fit_width) or (new_height!=fit_height):
             resized_image= pil_image.resize(fit_width,fit_height,Image.LANCZOS)
-            return resized_image
+            return np.array(resized_image)
         
-        return pil_image
+        return np.array(pil_image)
 
     output = process_img2img(image,prompt,strength,seed,inference_step)
    
@@ -142,7 +142,7 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                         id_input=gr.Text(label="Name", visible=False)
                             
                 with gr.Column():
-                    image_out = gr.Image(height=800,sources=[],label="Output", elem_id="output-img",format="jpg",type="pil")
+                    image_out = gr.Image(height=800,sources=[],label="Output", elem_id="output-img",format="jpg")