Update app.py

app.py

@@ -29,16 +29,16 @@ def split_image(input_image, num_splits=4):
 
     return output_images
 
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch_dtype)
-#pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
-#pipe.fuse_lora(lora_scale=0.125)
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch_dtype)
+pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-16steps-lora.safetensors"))
+pipe.fuse_lora(lora_scale=0.125)
 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)):
-    prompt_template = f"A side by side 4 frame image showing consecutive stills from a looped gif moving from left to right. The gif is {prompt}"
+    prompt_template = f"A side by side 4 frame image showing consecutive stills from a looped gif moving from left to right. The gif contains temporally coherent stills of {prompt}"
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
         
@@ -74,7 +74,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
         # FLUX Animations
-        Generate gifs with FLUX [dev] + HyperSD 8 steps. Concept idea by [fofr](https://x.com/fofrAI). Diffusers implementation by [Dhruv](_DhruvNair_)
+        Generate gifs with FLUX [dev] + HyperSD 16 steps. Concept idea by [fofr](https://x.com/fofrAI). Diffusers implementation by [Dhruv](_DhruvNair_)
         """)
         
         with gr.Row():
@@ -106,9 +106,9 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps = gr.Slider(
                     label="Number of inference steps",
                     minimum=1,
-                    maximum=12,
+                    maximum=32,
                     step=1,
-                    value=4, 
+                    value=16, 
             )