Refer image resize

app.py

@@ -24,17 +24,19 @@ MAX_IMAGE_SIZE = 1024
 @spaces.GPU
 def infer(prompt, refer_image, height, width):
 
+    init_image = refer_image.resize((1024, 1024))
+
     image = pipeline(
         prompt=prompt,
         # guidance_scale=guidance_scale,
         # num_inference_steps=num_inference_steps,
         width=width,
         height=height,
-        image=refer_image
+        image=init_image
         # generator=generator
     ).images[0]
 
-    return 'image'
+    return image
 
 
 examples = [