Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -35,6 +35,7 @@ if __name__ == "__main__":
|
|
35 |
metadata_dataset="coco_2017_val",
|
36 |
)
|
37 |
|
|
|
38 |
def inference(img, confidence):
|
39 |
img = np.array(img)
|
40 |
_, results = vis_demo.run_on_image(img, confidence)
|
|
|
35 |
metadata_dataset="coco_2017_val",
|
36 |
)
|
37 |
|
38 |
+
@spaces.GPU
|
39 |
def inference(img, confidence):
|
40 |
img = np.array(img)
|
41 |
_, results = vis_demo.run_on_image(img, confidence)
|