Update app.py

app.py

@@ -19,17 +19,11 @@ def split_image(input_image, num_splits=4):
     # Create a list to store the output images
     output_images = []
 
-    # Get image dimensions
-    img_width, img_height = input_image.size
-
-    # Calculate the width of each split
-    split_width = img_width // num_splits
-
-    # Split the image into equal sections horizontally
+    # Split the image into four 256x256 sections
     for i in range(num_splits):
-        left = i * split_width
-        right = (i + 1) * split_width
-        box = (left, 0, right, img_height)
+        left = i * 256
+        right = (i + 1) * 256
+        box = (left, 0, right, 256)
         output_images.append(input_image.crop(box))
 
     return output_images