run spaces on zero-GPU

app.py

@@ -1,5 +1,6 @@
 import json
 import random
+import spaces
 
 import gradio as gr
 import matplotlib.pyplot as plt
@@ -133,6 +134,7 @@ def draw_predictions(
     return imgs_list
 
 
+@spaces.GPU
 def inference(image, model_name, mask_threshold, bbox_threshold):
     """
     Load the ONNX model and run inference with the provided input `image`. Visualize the predictions and save them in a

requirements.txt

@@ -4,4 +4,5 @@ torchvision>=0.17.1
 matplotlib>=3.8.3
 numpy>=1.26.4
 pillow>=10.2.0
-gradio>=4.26.0
+gradio>=4.26.0
+spaces>=0.26.2