Update app.py

app.py

@@ -8,7 +8,7 @@ def list_flux_layers():
         # Load the FLUX.1-dev model
         # Using torch.bfloat16 to reduce memory usage, offloading to CPU if needed
         pipe = FluxPipeline.from_pretrained(
-            "black-forest-labs/FLUX.1-dev",
+            "black-forest-labs/FLUX.1-Kontext-dev",
             torch_dtype=torch.bfloat16
         )
         pipe.enable_model_cpu_offload()  # Offload to CPU to save VRAM