resize back 32

app.py

@@ -70,15 +70,15 @@ def process_images(image,prompt="a girl",strength=0.75,seed=0,inference_step=4,p
         output = pipe(prompt=prompt, image=image,generator=generator,strength=strength,width=width,height=height
                     ,guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
 
-        pil_image = Image.fromarray(output.images[0])
+        pil_image = output.images[0]#Image.fromarray()
         new_width,new_height = pil_image.size
 
         # 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 np.array(resized_image)
+            return resized_image
         
-        return np.array(pil_image)
+        return pil_image
 
     output = process_img2img(image,prompt,strength,seed,inference_step)