PawanratRung commited on
Commit
54376b3
·
verified ·
1 Parent(s): af0d355

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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()