Spaces:
Sleeping
Sleeping
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()
|