skylord commited on
Commit
2a31c0e
1 Parent(s): efb76b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def show_preds_image(image):
14
 
15
  image = cv2.imread(image_path)
16
  outputs = model.predict(source=image_path)
17
- print("output:: ", outputs)
18
  results = outputs[0].boxes.xyxy.cpu().numpy()
19
  for det in results:
20
  x1, y1, x2, y2 = det[:4]
 
14
 
15
  image = cv2.imread(image_path)
16
  outputs = model.predict(source=image_path)
17
+ print("output:: ", outputs[0].probs.cpu().numpy())
18
  results = outputs[0].boxes.xyxy.cpu().numpy()
19
  for det in results:
20
  x1, y1, x2, y2 = det[:4]