Spaces:
Running
on
Zero
Running
on
Zero
rizavelioglu
commited on
Commit
•
75b360c
1
Parent(s):
35ed39f
Update app.py
Browse files
app.py
CHANGED
@@ -156,7 +156,7 @@ 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 |
|