Update

README.md

@@ -4,7 +4,7 @@ emoji: 🐠
 colorFrom: purple
 colorTo: red
 sdk: gradio
-sdk_version: 3.17.0
+sdk_version: 4.31.5
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -5,6 +5,7 @@ import pathlib
 import gradio as gr
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 from sahi.prediction import ObjectPrediction
 from sahi.utils.cv import visualize_object_predictions
@@ -20,6 +21,7 @@ model = DetaForObjectDetection.from_pretrained(MODEL_ID)
 model.to(device)
 
 
+@spaces.GPU
 @torch.inference_mode()
 def run(image_path: str, threshold: float) -> np.ndarray:
     image = PIL.Image.open(image_path)
@@ -48,20 +50,22 @@ with gr.Blocks(css="style.css") as demo:
     with gr.Row():
         with gr.Column():
             image = gr.Image(label="Input image", type="filepath")
-            threshold = gr.Slider(label="Score threshold", minimum=0, maximum=1, value=0.1, step=0.01)
-            run_button = gr.Button("Run")
-        result = gr.Image(label="Result", type="numpy")
+            threshold = gr.Slider(label="Score threshold", minimum=0, maximum=1, step=0.01, value=0.1)
+            run_button = gr.Button()
+        result = gr.Image(label="Result")
+    gr.Examples(
+        examples=[[path, 0.1] for path in sorted(pathlib.Path("images").glob("*.jpg"))],
+        inputs=[image, threshold],
+        outputs=result,
+        fn=run,
+    )
 
-    with gr.Row():
-        paths = sorted(pathlib.Path("images").glob("*.jpg"))
-        gr.Examples(
-            examples=[[path.as_posix(), 0.1] for path in paths],
-            inputs=[image, threshold],
-            outputs=result,
-            fn=run,
-            cache_examples=True,
-        )
-
-    run_button.click(fn=run, inputs=[image, threshold], outputs=result)
+    run_button.click(
+        fn=run,
+        inputs=[image, threshold],
+        outputs=result,
+        api_name="predict",
+    )
 
-demo.queue().launch()
+if __name__ == "__main__":
+    demo.queue(max_size=20).launch()

requirements.txt

@@ -1,6 +1,8 @@
-git+https://github.com/huggingface/transformers@59d5ede
-numpy==1.24.1
-Pillow==9.4.0
-sahi==0.11.11
-torch==1.13.1
-torchvision==0.14.1
+gradio==4.31.5
+numpy==1.26.4
+Pillow==10.3.0
+sahi==0.11.16
+spaces==0.28.3
+torch==2.0.1
+torchvision==0.15.2
+transformers==4.41.1

style.css

@@ -1,3 +1,11 @@
 h1 {
   text-align: center;
+  display: block;
+}
+
+#duplicate-button {
+  margin: auto;
+  color: #fff;
+  background: #1565c0;
+  border-radius: 100vh;
 }