Update app.py

app.py

@@ -49,12 +49,12 @@ def infer(prompt, seed, randomize_seed, num_inference_steps, progress=gr.Progres
         num_inference_steps=num_inference_steps,
         num_images_per_prompt=1,
         generator=torch.Generator(device).manual_seed(seed),
-        height=height,
-        width=width
+        height=256,
+        width=1024
     ).images[0]
     gif_name = f"{uuid.uuid4().hex}-flux.gif"
     export_to_gif(split_image(image, 4), gif_name, fps=4)
-    return gif_name, seed
+    return gif_name, image, seed
 
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
@@ -73,8 +73,8 @@ with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # FLUX.1 Schnell Animations
-        Generate gifs with 
+        # FLUX Animations
+        Generate gifs with FLUX Schnell
         """)
         
         with gr.Row():
@@ -90,7 +90,7 @@ with gr.Blocks(css=css) as demo:
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
-
+        result_full = gr.Image(label="Gif Strip")
         with gr.Accordion("Advanced Settings", open=False):
             
             seed = gr.Slider(
@@ -120,7 +120,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn = infer,
         inputs = [prompt, seed, randomize_seed, num_inference_steps],
-        outputs = [result, seed]
+        outputs = [result, result_full, seed]
     )
 
 demo.queue().launch()