Use Flux.1

app.py

@@ -9,7 +9,7 @@ from diffusers import DiffusionPipeline
 import torch
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/stable-diffusion-3.5-large"  # Replace to the model you would like to use
+model_repo_id = "black-forest-labs/FLUX.1-schnell"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16
@@ -36,7 +36,7 @@ def infer(
     width=1024,
     height=1024,
     guidance_scale=4.5,
-    num_inference_steps=40,
+    num_inference_steps=4,
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed: