Update app.py

app.py

@@ -9,7 +9,7 @@ from diffusers import FluxTransformer2DModel
 from diffusers.utils import load_image
 from huggingface_hub import hf_hub_download
 
-kontext_path = hf_hub_download(repo_id="diffusers/kontext", filename="kontext.safetensors")
+kontext_path = hf_hub_download(repo_id="diffusers/kontext-v2", filename="dev-opt-2-a-3.safetensors")
 MAX_SEED = np.iinfo(np.int32).max
 transformer = FluxTransformer2DModel.from_single_file(kontext_path, torch_dtype=torch.bfloat16)
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", transformer=transformer, torch_dtype=torch.bfloat16).to("cuda")