update resizing

app.py

@@ -58,6 +58,11 @@ def chat(image_in, in_steps, in_guidance_scale, in_img_guidance_scale, image_hid
     else:
       seed = random.randint(0, 1000000)
       img_name = f"./edited_image_{seed}.png"
+      #Resizing to handle memory errors
+      basewidth = 512
+      wpercent = (basewidth/float(image_in.size[0]))
+      hsize = int((float(image_in.size[1])*float(wpercent)))
+      image_in = image_in.resize((basewidth,hsize), Image.Resampling.LANCZOS)
       edited_image = pipe(prompt, image=image_in, num_inference_steps=int(in_steps), guidance_scale=float(in_guidance_scale), image_guidance_scale=float(in_img_guidance_scale)).images[0]
       if os.path.exists(img_name):
         os.remove(img_name)