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 * 256
-        right = (i + 1) * 256
-        box = (left, 0, right, 256)
+        left = i * 384
+        right = (i + 1) * 384
+        box = (left, 0, right, 384)
         output_images.append(input_image.crop(box))
 
     return output_images
@@ -37,7 +37,7 @@ pipe.to(device=device, dtype=torch_dtype)
 MAX_SEED = np.iinfo(np.int32).max
 
 @spaces.GPU 
-def infer(prompt, seed=42, randomize_seed=False, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
+def infer(prompt, seed=1, randomize_seed=False, num_inference_steps=4, progress=gr.Progress(track_tqdm=True)):
     prompt_template = f"A side by side 4 frame image showing consecutive stills from a looped gif moving from left to right. The stills are of {prompt}"
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -49,8 +49,8 @@ def infer(prompt, seed=42, 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=256,
-        width=1024
+        height=384,
+        width=1152
     ).images[0]
     gif_name = f"{uuid.uuid4().hex}-flux.gif"
     export_to_gif(split_image(image, 4), gif_name, fps=4)