Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -156,7 +156,11 @@ def inference(image, model_name, mask_threshold, bbox_threshold):
|
|
| 156 |
# Create an inference session.
|
| 157 |
ort_session = onnxruntime.InferenceSession(
|
| 158 |
path_onnx,
|
| 159 |
-
providers=["CPUExecutionProvider"],
|
|
|
|
|
|
|
|
|
|
|
|
|
| 160 |
sess_options=sess_options,
|
| 161 |
)
|
| 162 |
|
|
|
|
| 156 |
# Create an inference session.
|
| 157 |
ort_session = onnxruntime.InferenceSession(
|
| 158 |
path_onnx,
|
| 159 |
+
providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
|
| 160 |
+
sess_options=sess_options,
|
| 161 |
+
)
|
| 162 |
+
|
| 163 |
+
|
| 164 |
sess_options=sess_options,
|
| 165 |
)
|
| 166 |
|