update app

app.py

@@ -99,19 +99,15 @@ if torch.cuda.is_available():
 
 print("Using device:", device)
 
-# --- Main Model Initialization ---
 MAX_SEED = np.iinfo(np.int32).max
 pipe = FluxKontextPipeline.from_pretrained("black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.bfloat16).to("cuda")
 
-# --- Load All Adapters ---
-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")
-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-Top-Down-View", weight_name="Kontext-Top-Down-View.safetensors", adapter_name="top-down")
+pipe.load_lora_weights("prithivMLmods/Kontext-Bottom-Up-View", weight_name="Kontext-Bottom-Up-View.safetensors", adapter_name="bottom-up")
+pipe.load_lora_weights("prithivMLmods/Kontext-CAM-Left-View", weight_name="Kontext-CAM-Left-View.safetensors", adapter_name="left-view")
+pipe.load_lora_weights("prithivMLmods/Kontext-CAM-Right-View", weight_name="Kontext-CAM-Right-View.safetensors", adapter_name="right-view")
+pipe.load_lora_weights("starsfriday/Kontext-Remover-General-LoRA", weight_name="kontext_remove.safetensors", adapter_name="kontext-remove")
 
-# --- Upscaler Model Initialization ---
 aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
 
 @spaces.GPU
@@ -122,18 +118,16 @@ def infer(input_image, prompt, lora_adapter, upscale_image, seed=42, randomize_s
     if not input_image:
         raise gr.Error("Please upload an image for editing.")
 
-    if lora_adapter == "PhotoCleanser":
-        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])
-    elif lora_adapter == "MonochromePencil":
-        pipe.set_adapters(["pencil"], adapter_weights=[1.0])
-    elif lora_adapter == "LZO-Zoom":
-        pipe.set_adapters(["lzo"], adapter_weights=[1.0])
-    elif lora_adapter == "Kontext-Watermark-Remover":
-        pipe.set_adapters(["watermark-remover"], adapter_weights=[1.0])
+    if lora_adapter == "Kontext-Top-Down-View":
+        pipe.set_adapters(["top-down"], adapter_weights=[1.0])
+    elif lora_adapter == "Kontext-Bottom-Up-View":
+        pipe.set_adapters(["bottom-up"], adapter_weights=[1.0])
+    elif lora_adapter == "Kontext-CAM-Left-View":
+        pipe.set_adapters(["left-view"], adapter_weights=[1.0])
+    elif lora_adapter == "Kontext-CAM-Right-View":
+        pipe.set_adapters(["right-view"], adapter_weights=[1.0])
+    elif lora_adapter == "Kontext-Remover":
+        pipe.set_adapters(["kontext-remove"], adapter_weights=[1.0])
         
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -221,12 +215,12 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
             with gr.Column():
                 output_slider = ImageSlider(label="Before / After", show_label=False, interactive=False)
                 reuse_button = gr.Button("Reuse this image", visible=False)
-                
+
                 with gr.Row():
                     lora_adapter = gr.Dropdown(
                         label="Chosen LoRA",
-                        choices=["PhotoCleanser", "PhotoRestorer", "PolaroidWarm", "MonochromePencil", "LZO-Zoom", "Kontext-Watermark-Remover"],
-                        value="PhotoCleanser"
+                        choices=["Kontext-Top-Down-View", "Kontext-Remover", "Kontext-Bottom-Up-View", "Kontext-CAM-Left-View", "Kontext-CAM-Right-View"],
+                        value="Kontext-Top-Down-View"
                     )
                     
                 with gr.Row():
@@ -234,14 +228,11 @@ with gr.Blocks(css=css, theme=orange_red_theme) as demo:
 
         gr.Examples(
             examples=[
-                ["photocleanser/2.png", "[photo content], remove the cat from the image while preserving the background and remaining elements, maintaining realism and original details.", "PhotoCleanser"],
-                ["photocleanser/1.png", "[photo content], remove the football from the image while preserving the background and remaining elements, maintaining realism and original details.", "PhotoCleanser"],
-                ["watermark/12.jpeg", "[photo content], remove any watermark text or logos from the image while preserving the background, texture, lighting, and overall realism. Ensure the edited areas blend seamlessly with surrounding details, leaving no visible traces of watermark removal.", "Kontext-Watermark-Remover"],
-                ["photorestore/1.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"],
-                ["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"],
-                ["pencil/1.png", "[photo content], replicate the image as a pencil illustration, black and white, with sketch-like detailing.", "MonochromePencil"],
+                ["examples/1.png", "[photo content], recreate the scene from a top-down perspective. Maintain all visual proportions, lighting consistency, and realistic spatial relationships. Ensure the background, textures, and environmental shadows remain naturally aligned from this elevated angle.", "PhotoCleanser"],
+                ["examples/2.png", "[photo content], remove the football from the image while preserving the background and remaining elements, maintaining realism and original details.", "PhotoCleanser"],
+                ["examples/3.jpeg", "[photo content], remove any watermark text or logos from the image while preserving the background, texture, lighting, and overall realism. Ensure the edited areas blend seamlessly with surrounding details, leaving no visible traces of watermark removal.", "Kontext-Watermark-Remover"],
+                ["examples/4.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"],
+                ["examples/5.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"],
             ],
             inputs=[input_image, prompt, lora_adapter],
             outputs=[output_slider, seed],