panda1835 commited on
Commit
cf34237
·
1 Parent(s): fe79314

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -11,12 +11,11 @@ from ultralytics import YOLO
11
  detect_model = YOLO('yolov8m_2023-10-23_best.pt')
12
 
13
  def detect(image):
14
- # Inference
15
- results = detect_model(image)
16
  try:
17
  results = results[0].boxes.xyxy[0].cpu().numpy()
18
- top = int(results[0])
19
- left = int(results[1])
20
  width = int(results[2] - results[0])
21
  height = int(results[3] - results[1])
22
  return {
 
11
  detect_model = YOLO('yolov8m_2023-10-23_best.pt')
12
 
13
  def detect(image):
14
+ results = detect_model(image, conf=0.5)
 
15
  try:
16
  results = results[0].boxes.xyxy[0].cpu().numpy()
17
+ top = int(results[1])
18
+ left = int(results[0])
19
  width = int(results[2] - results[0])
20
  height = int(results[3] - results[1])
21
  return {