Update app.py

app.py

@@ -22,9 +22,9 @@ def split_image(input_image, num_splits=4):
 
     # Split the image into four 256x256 sections
     for i in range(num_splits):
-        left = i * 288
-        right = (i + 1) * 288
-        box = (left, 0, right, 288)
+        left = i * 320
+        right = (i + 1) * 320
+        box = (left, 0, right, 320)
         output_images.append(input_image.crop(box))
 
     return output_images
@@ -49,8 +49,8 @@ def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=4, progress=
         num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         generator=torch.Generator(device).manual_seed(seed),
-        height=384,
-        width=1152
+        height=320,
+        width=1280
     ).images[0]
     gif_name = f"{uuid.uuid4().hex}-flux.gif"
     export_to_gif(split_image(image, 4), gif_name, fps=4)