added image resizing

app.py

@@ -42,7 +42,25 @@ def remove_bg(image: Image):
     # Apply background removal using rembg
     output_array_bg = rembg.remove(input_array_bg)
     # Create a PIL Image from the output array
-    return Image.fromarray(output_array_bg)
+    img = Image.fromarray(output_array_bg)
+
+    mask = img.convert('L')  # Convert to grayscale
+    mask_array = np.array(mask)
+
+    # Create a binary mask (non-background areas are 255, background areas are 0)
+    binary_mask = mask_array > 0
+
+    # Find the bounding box of the non-background areas
+    coords = np.argwhere(binary_mask)
+    x0, y0 = coords.min(axis=0)
+    x1, y1 = coords.max(axis=0) + 1
+
+    # Crop the output image using the bounding box
+    cropped_output_image = img.crop((y0, x0, y1, x1))
+
+    # Resize the cropped image to 1024x1024
+    upscaled_image = cropped_output_image.resize((1024, 1024), Image.LANCZOS)
+    return upscaled_image
     
 def update_selection(evt: gr.SelectData):
     selected_lora = loras[evt.index]