ysr commited on
Commit
36908fb
·
1 Parent(s): df051f3

change a slight

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ def show_preds_image(image_path):
10
  image = cv2.imread(image_path)
11
  outputs = model.predict(source=image_path)
12
  print("output:", outputs)
13
- results = outputs[0].cpu().numpy()
14
  print("results:", results)
15
 
16
  for i, det in enumerate(results.boxes.xyxy):
 
10
  image = cv2.imread(image_path)
11
  outputs = model.predict(source=image_path)
12
  print("output:", outputs)
13
+ results = outputs[0]
14
  print("results:", results)
15
 
16
  for i, det in enumerate(results.boxes.xyxy):