Spaces:
Running
on
Zero
Running
on
Zero
rizavelioglu
commited on
Commit
•
e044f2c
1
Parent(s):
62e7cee
specify estimated duration of GPU usage
Browse files
app.py
CHANGED
|
@@ -134,7 +134,7 @@ def draw_predictions(
|
|
| 134 |
return imgs_list
|
| 135 |
|
| 136 |
|
| 137 |
-
@spaces.GPU
|
| 138 |
def inference(image, model_name, mask_threshold, bbox_threshold):
|
| 139 |
"""
|
| 140 |
Load the ONNX model and run inference with the provided input `image`. Visualize the predictions and save them in a
|
|
|
|
| 134 |
return imgs_list
|
| 135 |
|
| 136 |
|
| 137 |
+
@spaces.GPU(duration=30)
|
| 138 |
def inference(image, model_name, mask_threshold, bbox_threshold):
|
| 139 |
"""
|
| 140 |
Load the ONNX model and run inference with the provided input `image`. Visualize the predictions and save them in a
|