Update app.py

app.py

@@ -13,10 +13,11 @@ 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 all three adapters to the app with unique names
+# Add all four 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")
+pipe.load_lora_weights("prithivMLmods/Monochrome-Pencil", weight_name="Monochrome-Pencil-i2i.safetensors", adapter_name="pencil")
 
 
 @spaces.GPU
@@ -47,6 +48,8 @@ def infer(input_image, prompt, lora_adapter, seed=42, randomize_seed=False, guid
         pipe.set_adapters(["restorer"], adapter_weights=[1.0])
     elif lora_adapter == "PolaroidWarm":
         pipe.set_adapters(["polaroid"], adapter_weights=[1.0])
+    elif lora_adapter == "MonochromePencil":
+        pipe.set_adapters(["pencil"], adapter_weights=[1.0])
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -95,11 +98,6 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             with gr.Column():
                 input_image = gr.Image(label="Upload the image for editing", type="pil", height="300")
-                lora_adapter = gr.Dropdown(
-                    label="Choose Adapter",
-                    choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm"],
-                    value="PhotoCleanser"
-                )
                 with gr.Row():
                     prompt = gr.Text(
                         label="Prompt",
@@ -140,7 +138,13 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 result = gr.Image(label="Result", show_label=False, interactive=False, format="png")
                 reuse_button = gr.Button("Reuse this image", visible=False)
-        
+                with gr.Row():
+                    lora_adapter = gr.Dropdown(
+                    label="Choose Adapter",
+                    choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm", "MonochromePencil"],
+                    value="PhotoCleanser"
+                )
+                    
         # Consolidate all examples into a single component with a new "Selected LoRA" column
         gr.Examples(
             examples=[
@@ -173,6 +177,16 @@ with gr.Blocks(css=css) as demo:
                     "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"
+                ],
+                [
+                    "pencil/1.png",
+                    "[photo content], transform the image into a detailed monochrome pencil sketch, emphasizing sharp lines, textures, and shading for a classic hand-drawn look.",
+                    "MonochromePencil"
+                ],
+                [
+                    "pencil/2.png",
+                    "[photo content], convert the photo into a realistic graphite pencil drawing, capturing the subject's form and depth with varied strokes and contrast.",
+                    "MonochromePencil"
                 ]
             ],
             # The inputs now include the lora_adapter dropdown