Update app.py

app.py

@@ -10,7 +10,7 @@ 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")
+pipe = FluxKontextPipeline.from_pretrained("prithivMLmods/Flux.1-Fusion-KonKrea", torch_dtype=torch.bfloat16).to("cuda")
 
 @spaces.GPU
 def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5, steps=28, progress=gr.Progress(track_tqdm=True)):
@@ -91,8 +91,8 @@ css="""
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""# FLUX.1 Kontext [dev]
-Image editing and manipulation model guidance-distilled from FLUX.1 Kontext [pro], [[blog]](https://bfl.ai/announcements/flux-1-kontext-dev) [[model]](https://huggingface.co/black-forest-labs/FLUX.1-Kontext-dev)
+        gr.Markdown(f"""# Flux.1 Fusion KonKrea
+Image editing and manipulation model guidance-distilled from Flux.1-Fusion-KonKrea, [[model]](https://huggingface.co/prithivMLmods/Flux.1-Fusion-KonKrea)
         """)
         with gr.Row():
             with gr.Column():