bump deps

README.md

@@ -4,8 +4,8 @@ emoji: ✂️
 colorFrom: indigo
 colorTo: indigo
 sdk: gradio
-sdk_version: 5.27.1
-python_version: 3.12
+sdk_version: 6.20.0
+python_version: 3.12.12
 app_file: src/app.py
 pinned: false
 license: mit

pyproject.toml

@@ -6,10 +6,11 @@ authors = [
     { name = "Pierre Chapuis", email = "pierre@lagon.tech" }
 ]
 dependencies = [
-    "gradio>=5.27.1",
-    "gradio-image-annotation>=0.4.1,<0.5.0",
+    "gradio>=6.20.0",
+    "gradio-image-annotation>=0.4.1",
     "pillow>=11.3.0",
     "pi-heif>=1.1.0",
+    "torch==2.11.0",
     "refiners @ git+https://github.com/finegrain-ai/refiners",
     "numba>=0.60.0",
     "pymatting>=1.1.12",

requirements.txt

@@ -1,7 +1,8 @@
 spaces>=0.35.0
-gradio_image_annotation>=0.4.1,<0.5.0
+gradio_image_annotation>=0.4.1
 pillow>=11.3.0
 pi_heif>=1.1.0
+torch==2.11.0
 git+https://github.com/finegrain-ai/refiners@cfe8b66ba4f8a906583850ac25e9e89cb83a44b9
 numba>=0.60.0
 pymatting>=1.1.12

src/app.py

@@ -56,7 +56,9 @@ def gd_detect(img: Image.Image, prompt: str) -> BoundingBox | None:
     assert isinstance(gd_processor, GroundingDinoProcessor)
 
     # Grounding Dino expects a dot after each category.
-    inputs = gd_processor(images=img, text=f"{prompt}.", return_tensors="pt").to(device=device)
+    # transformers 5 accepts return_tensors at runtime but omits it from the typed processor kwargs.
+    processor_kwargs: dict[str, Any] = {"return_tensors": "pt"}
+    inputs = gd_processor(images=img, text=f"{prompt}.", **processor_kwargs).to(device=device)
 
     with no_grad():
         outputs = gd_model(**inputs)
@@ -167,7 +169,7 @@ def process_bbox(prompts: dict[str, Any]) -> tuple[tuple[Image.Image, Image.Imag
     return _process(img, bbox)
 
 
-def on_change_bbox(prompts: dict[str, Any] | None):
+def on_change_bbox(prompts: dict[str, Any] | None) -> dict[str, Any]:
     return gr.update(interactive=prompts is not None)
 
 
@@ -175,7 +177,7 @@ def process_prompt(img: Image.Image, prompt: str) -> tuple[tuple[Image.Image, Im
     return _process(img, prompt)
 
 
-def on_change_prompt(img: Image.Image | None, prompt: str | None):
+def on_change_prompt(img: Image.Image | None, prompt: str | None) -> dict[str, Any]:
     return gr.update(interactive=bool(img and prompt))
 
 
@@ -184,9 +186,11 @@ TITLE = """
 <p>
     Create HD cutouts from any image with just a prompt.
 </p><p>
-    This Space showcases one of Finegrain's earlier generative AI models. Today, we're building the Agentic Camera: an on-device system that produces the intended photo at capture time, eliminating image editing in most cases.
+    This Space showcases one of Finegrain's earlier generative AI models. Today, we're building the Agentic Camera:
+    an on-device system that produces the intended photo at capture time, eliminating image editing in most cases.
 </p><p>
-    ✨ Learn more at <a href="https://finegrain.ai">finegrain.ai</a> or follow <a href="https://huggingface.co/finegrain">Finegrain</a> for updates.
+    ✨ Learn more at <a href="https://finegrain.ai">finegrain.ai</a>
+    or follow <a href="https://huggingface.co/finegrain">Finegrain</a> for updates.
 </p>
 """
 
@@ -201,7 +205,7 @@ with gr.Blocks() as demo:
                 prompt = gr.Textbox(label="What should we cut?")
                 btn = gr.ClearButton(value="Cut Out Object", interactive=False)
             with gr.Column():
-                oimg = gr.ImageSlider(label="Before / After", max_height=1500, show_fullscreen_button=False)
+                oimg = gr.ImageSlider(label="Before / After", max_height=1500, buttons=["download"])
                 dlbt = gr.DownloadButton("Download Cutout", interactive=False)
 
         btn.add(oimg)
@@ -255,7 +259,7 @@ with gr.Blocks() as demo:
                 )
                 btn = gr.ClearButton(value="Cut Out Object", interactive=False)
             with gr.Column():
-                oimg = gr.ImageSlider(label="Before / After", max_height=1500, show_fullscreen_button=False)
+                oimg = gr.ImageSlider(label="Before / After", max_height=1500, buttons=["download"])
                 dlbt = gr.DownloadButton("Download Cutout", interactive=False)
 
         btn.add(oimg)