Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -159,11 +159,8 @@ def inference(image, model_name, mask_threshold, bbox_threshold):
|
|
159 |
providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
|
160 |
sess_options=sess_options,
|
161 |
)
|
162 |
-
|
163 |
-
|
164 |
sess_options=sess_options,
|
165 |
)
|
166 |
-
|
167 |
# compute ONNX Runtime output prediction
|
168 |
ort_inputs = {
|
169 |
ort_session.get_inputs()[0].name: img_transformed.unsqueeze(dim=0).numpy()
|
|
|
159 |
providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
|
160 |
sess_options=sess_options,
|
161 |
)
|
|
|
|
|
162 |
sess_options=sess_options,
|
163 |
)
|
|
|
164 |
# compute ONNX Runtime output prediction
|
165 |
ort_inputs = {
|
166 |
ort_session.get_inputs()[0].name: img_transformed.unsqueeze(dim=0).numpy()
|