Handle images

app.py

@@ -585,9 +585,9 @@ def generate_image(input_image_path, prompt):
 
     resized_input_img = resize_to_square(input_image_path, 512)
 
-    resized_input_img_path = resized_input_img.save('input_img.png')
+    # resized_input_img_path = resized_input_img.save('input_img.png')
 
-    ai_gen_image = generate_ai_bg(resized_input_img_path, prompt)
+    ai_gen_image = generate_ai_bg(input_image_path, prompt)
 
     mask_input_image = run_grounded_sam(resized_input_img)