Upscale images

app.py

@@ -441,6 +441,10 @@ def resize_image(image, scaling_factor=1):
         int(image.height * scaling_factor)))
     return image
 
+def upscale_image(image, size):
+    image.resize((size, size), Image.LANCZOS)
+    return image
+
 def resize_to_square(image, size=1024):
 
     # Load image if a file path is provided
@@ -598,11 +602,15 @@ def generate_image(input_image_path, prompt):
 
     ai_gen_image = generate_ai_bg(resized_input_img, prompt)
 
-    mask_input_image = run_grounded_sam(resized_input_img)
+    ai_gen_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(resized_input_img, ai_gen_image, mask_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 = resize_to_square(final_image, 1024)
 
     return final_image