Update app.py

app.py

@@ -28,11 +28,10 @@ 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.to(device)
+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-8steps-lora.safetensors"))
+pipe.fuse_lora(lora_scale=0.125)
+pipe.to(device=device, dtype=torch_dtype)
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -74,7 +73,7 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
         # FLUX Animations
-        Generate gifs with FLUX Schnell
+        Generate gifs with FLUX [dev] + HyperSD 8 steps. Concept idea by [fofr](https://x.com/fofrAI). Diffusers implementation by [Dhruv](_DhruvNair_)
         """)
         
         with gr.Row():
@@ -108,7 +107,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=1,
                     maximum=12,
                     step=1,
-                    value=4, 
+                    value=8, 
             )