Upscale image properly

app.py

@@ -442,8 +442,8 @@ def resize_image(image, scaling_factor=1):
     return image
 
 def upscale_image(image, size):
-    image.resize((size, size), Image.LANCZOS)
-    return image
+    new_image = image.resize((size, size), Image.LANCZOS)
+    return new_image
 
 def resize_to_square(image, size=1024):
 
@@ -602,15 +602,13 @@ def generate_image(input_image_path, prompt):
 
     ai_gen_image = generate_ai_bg(resized_input_img, prompt)
 
-    ai_gen_image = upscale_image(ai_gen_image, 8192)
+    upscaled_ai_image = upscale_image(ai_gen_image, 8192)
 
     upscaled_input_image = upscale_image(resized_input_img, 8192)
 
     mask_input_image = run_grounded_sam(upscaled_input_image)
 
-    final_image = blend_details(upscaled_input_image, ai_gen_image, mask_input_image)
-
-    # final_image = resize_to_square(final_image, 1024)
+    final_image = blend_details(upscaled_input_image, upscaled_ai_image, mask_input_image)
 
     return final_image