Update app.py

app.py

@@ -6,12 +6,22 @@ import random
 from PIL import Image
 
 from diffusers import FluxKontextPipeline
+from diffusers.models import FluxTransformer2DModel
+
 from diffusers.utils import load_image
 
 MAX_SEED = np.iinfo(np.int32).max
 
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 
+# New Here
+turbo = FluxTransformer2DModel.from_pretrained(
+    "Yuanshi/FLUX.1-Kontext-Turbo", subfolder="transformer", torch_dtype=torch.bfloat16
+).to("cuda")
+pipe.transformer = turbo
+
+print("Model dev loaded.")
+
 @spaces.GPU
 def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
     """
@@ -130,7 +140,7 @@ Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro
                         label="Steps",
                         minimum=1,
                         maximum=30,
-                        value=28,
+                        value=8,
                         step=1
                     )