Spaces:
Running
Running
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 |
|