update app

app.py

@@ -38,6 +38,7 @@ pipe.load_lora_weights("prithivMLmods/Polaroid-Warm-i2i", weight_name="Polaroid-
 pipe.load_lora_weights("prithivMLmods/Monochrome-Pencil", weight_name="Monochrome-Pencil-i2i.safetensors", adapter_name="pencil")
 pipe.load_lora_weights("prithivMLmods/LZO-1-Preview", weight_name="LZO-1-Preview.safetensors", adapter_name="lzo")
 pipe.load_lora_weights("prithivMLmods/Kontext-Watermark-Remover", weight_name="Kontext-Watermark-Remover.safetensors", adapter_name="watermark-remover")
+pipe.load_lora_weights("prithivMLmods/Kontext-Unblur-Upscale", weight_name="Kontext-Image-Upscale.safetensors", adapter_name="unblur-upscale")
 
 # --- Upscaler Model Initialization ---
 aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
@@ -62,6 +63,8 @@ def infer(input_image, prompt, lora_adapter, upscale_image, seed=42, randomize_s
         pipe.set_adapters(["lzo"], adapter_weights=[1.0])
     elif lora_adapter == "Kontext-Watermark-Remover":
         pipe.set_adapters(["watermark-remover"], adapter_weights=[1.0])
+    elif lora_adapter == "Kontext-Unblur-Upscale":
+        pipe.set_adapters(["unblur-upscale"], adapter_weights=[1.0])
         
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -100,7 +103,7 @@ css="""
 #main-title h1 {font-size: 2.1em !important;}
 """
 
-with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
+with gr.Blocks(css=css, theme="allenai/gradio-theme") as demo:
     
     with gr.Column(elem_id="col-container"):
         gr.Markdown("# **Photo-Mate-i2i**", elem_id="main-title")
@@ -153,7 +156,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 with gr.Row():
                     lora_adapter = gr.Dropdown(
                         label="Chosen LoRA",
-                        choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm", "MonochromePencil", "LZO-Zoom", "Kontext-Watermark-Remover"],
+                        choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm", "MonochromePencil", "LZO-Zoom", "Kontext-Watermark-Remover", "Kontext-Unblur-Upscale"],
                         value="PhotoCleanser"
                     )
                     
@@ -169,7 +172,11 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 ["lzo/1.jpg", "[photo content], zoom in on the specified [face close-up], enhancing resolution and detail while preserving sharpness, realism, and original context. Maintain natural proportions and background continuity around the zoomed area.", "LZO-Zoom"],
                 ["photorestore/2.png", "[photo content], restore and enhance the image by repairing any damage, scratches, or fading. Colorize the photo naturally while preserving authentic textures and details, maintaining a realistic and historically accurate look.", "PhotoRestorer"],
                 ["polaroid/1.png", "[photo content], in the style of a vintage Polaroid, with warm, faded tones, and a white border.", "PolaroidWarm"],
+                ["unblur/1.jpg", "[photo content], upscale the low-quality image to 4K resolution, enhancing sharpness, clarity, and fine details while preserving the original texture, colors, lighting, and natural appearance. Remove noise, blur, and compression artifacts without over-smoothing or distorting facial or object features. Ensure realistic depth, balanced contrast, and accurate tones, achieving a high-definition, lifelike result that maintains the integrity of the original image.", "Kontext-Unblur-Upscale"],
                 ["pencil/1.png", "[photo content], replicate the image as a pencil illustration, black and white, with sketch-like detailing.", "MonochromePencil"],
+                ["unblur/11.jpg", "[photo content], upscale the low-quality image to 4K resolution, enhancing sharpness, clarity, and fine details while preserving the original texture, colors, lighting, and natural appearance. Remove noise, blur, and compression artifacts without over-smoothing or distorting facial or object features. Ensure realistic depth, balanced contrast, and accurate tones, achieving a high-definition, lifelike result that maintains the integrity of the original image.", "Kontext-Unblur-Upscale"],
+
+            
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_slider, seed],