ZeroGPU @spaces.GPU

app.py

@@ -18,6 +18,8 @@ except ImportError:
     os.system('pip install transformers')
     os.system('pip install git+https://github.com/IDEA-Research/Grounded-Segment-Anything.git')
 
+import os
+import spaces
 import gradio as gr
 import json
 import numpy as np
@@ -26,6 +28,7 @@ from yolo_utils import yolo_processimage
 from detectron_utils import detectron_process_image
 from gsl_utils import gsl_process_image
 
+@spaces.GPU
 def process_image(image, include_json):
     detectron_result = detectron_process_image(image)
     yolo_result = yolo_processimage(image)
@@ -54,4 +57,4 @@ gr.Interface(
     ],
     title="Insect Model Zoo 🐞🔬",
     examples=examples
-).launch()
+).launch()