Update app.py

app.py

@@ -13,9 +13,10 @@ MAX_SEED = np.iinfo(np.int32).max
 
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 
-# Add both adapters to the app with unique names
+# Add all three adapters to the app with unique names
 pipe.load_lora_weights("prithivMLmods/PhotoCleanser-i2i", weight_name="PhotoCleanser-i2i.safetensors", adapter_name="cleanser")
 pipe.load_lora_weights("prithivMLmods/Photo-Restore-i2i", weight_name="Photo-Restore-i2i.safetensors", adapter_name="restorer")
+pipe.load_lora_weights("prithivMLmods/Polaroid-Warm-i2i", weight_name="Polaroid-Warm-i2i.safetensors", adapter_name="polaroid")
 
 
 @spaces.GPU
@@ -29,7 +30,7 @@ def infer(input_image, prompt, lora_adapter, seed=42, randomize_seed=False, guid
     Args:
         input_image (PIL.Image.Image): The input image to be edited.
         prompt (str): Text description of the desired edit.
-        lora_adapter (str): The name of the LoRA adapter to use ("PhotoCleanser" or "PhotoRestorer").
+        lora_adapter (str): The name of the LoRA adapter to use.
         seed (int, optional): Random seed for reproducible generation. Defaults to 42.
         randomize_seed (bool, optional): If True, generates a random seed. Defaults to False.
         guidance_scale (float, optional): Controls adherence to the prompt. Defaults to 2.5.
@@ -44,6 +45,8 @@ def infer(input_image, prompt, lora_adapter, seed=42, randomize_seed=False, guid
         pipe.set_adapters(["cleanser"], adapter_weights=[1.0])
     elif lora_adapter == "PhotoRestorer":
         pipe.set_adapters(["restorer"], adapter_weights=[1.0])
+    elif lora_adapter == "PolaroidWarm":
+        pipe.set_adapters(["polaroid"], adapter_weights=[1.0])
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -94,7 +97,7 @@ with gr.Blocks(css=css) as demo:
                 input_image = gr.Image(label="Upload the image for editing", type="pil", height="300")
                 lora_adapter = gr.Dropdown(
                     label="Choose Adapter",
-                    choices=["PhotoCleanser", "PhotoRestorer"],
+                    choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm"],
                     value="PhotoCleanser"
                 )
                 with gr.Row():
@@ -160,6 +163,16 @@ with gr.Blocks(css=css) as demo:
                     "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], apply a warm, vintage Polaroid-style filter, enhancing the image with nostalgic tones, soft focus, and characteristic light leaks for an authentic, retro feel.",
+                    "PolaroidWarm"
+                ],
+                [
+                    "polaroid/2.png",
+                    "[photo content], give the image a classic Polaroid look with warm, saturated colors, gentle fading, and a subtle vignette effect, evoking a sense of timeless memories.",
+                    "PolaroidWarm"
                 ]
             ],
             # The inputs now include the lora_adapter dropdown