Upload 12 files

.gitattributes

@@ -33,3 +33,12 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/1.jpg filter=lfs diff=lfs merge=lfs -text
+examples/10.jpg filter=lfs diff=lfs merge=lfs -text
+examples/2.jpg filter=lfs diff=lfs merge=lfs -text
+examples/3.jpg filter=lfs diff=lfs merge=lfs -text
+examples/4.jpg filter=lfs diff=lfs merge=lfs -text
+examples/6.jpg filter=lfs diff=lfs merge=lfs -text
+examples/7.jpg filter=lfs diff=lfs merge=lfs -text
+examples/8.jpg filter=lfs diff=lfs merge=lfs -text
+examples/9.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,111 @@
+from pathlib import Path
+
+import gradio as gr
+import pi_heif
+import spaces
+import requests
+import base64
+import io
+import os
+import json
+from huggingface_hub import hf_hub_download
+from PIL import Image
+
+css = """
+footer {display: none !important;}
+"""
+
+TITLE = """
+<p>
+    Erase unwanted watermarks automatically with AI, no manual editing needed.
+</p><p>
+    For premium-quality results,
+    <a href="https://nowatermark.cloud">try NoWatermark API</a>
+    — it's free to test!
+</p>
+"""
+
+HEADERS = os.getenv("HEADERS")
+API_HOST = os.getenv("API_HOST")
+
+
+def remove_watermark(img_bytes):
+    headers = json.loads(HEADERS)
+
+    files = {
+        "original_preview_image": ("1.webp", img_bytes, "image/webp"),
+    }
+
+    data = {
+        "zoom_factor": 1,
+        "predict_mode": "old",
+    }
+
+    resp = requests.post(
+        API_HOST,
+        headers=headers,
+        files=files,
+        data=data,
+    )
+    if resp.status_code != 200:
+        print(f"Error: {resp.status_code} - {resp.text}")
+        return None
+
+    base64_image = resp.json().get("edited_image").get("image")
+    return base64.b64decode(base64_image)
+
+def process(input_image):
+    """
+    input_image: PIL.Image from Gradio
+    returns: PIL.Image (dewatermarked)
+    """
+    # Convert PIL image to bytes (WebP, since your API expects webp)
+    img_bytes_io = io.BytesIO()
+    input_image.save(img_bytes_io, format="WEBP")
+    img_bytes = img_bytes_io.getvalue()
+
+    # Call your dewatermark function
+    result_bytes = remove_watermark(img_bytes)
+    if result_bytes is None:
+        return [input_image, input_image]
+        # return None  # or return input_image as fallback
+
+    # Convert bytes back to PIL image
+    result_img = Image.open(io.BytesIO(result_bytes)).convert("RGB")
+    return [input_image, result_img]
+    
+with gr.Blocks(css=css) as demo:
+    gr.HTML(TITLE)
+
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Input Image")
+            run_button = gr.ClearButton(components=None, value="Remove Watermark")
+        with gr.Column():
+            output_slider = gr.ImageSlider(label="Before / After", max_height=1200, show_fullscreen_button=False)
+            run_button.add(output_slider)
+
+    run_button.click(fn=process, inputs=input_image, outputs=output_slider, api_name=False)
+
+    gr.Examples(
+        examples=[
+            "examples/1.jpg",
+            "examples/2.jpg",
+            "examples/3.jpg",
+            "examples/4.jpg",
+            "examples/5.jpg",
+            "examples/6.jpg",
+            "examples/7.jpg",
+            "examples/8.jpg",
+            "examples/9.jpg",
+            "examples/10.jpg",
+        ],
+        inputs=[input_image],
+        outputs=output_slider,
+        fn=process,
+        cache_examples=True,
+        cache_mode="lazy",
+        run_on_click=False,
+    )
+
+demo.launch(share=False, ssr_mode=False, show_api=False)

requirements.txt

@@ -0,0 +1,5 @@
+spaces>=0.35.0
+gradio_imageslider==0.0.20
+numpy<2.0.0
+pillow>=11.3.0
+pi_heif>=1.1.0