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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -14,11 +14,11 @@ def detect(image):
14
  # Inference
15
  results = detect_model(image)
16
  try:
17
- results = json.loads(results.pandas().xyxy[0].to_json(orient="records"))[0]
18
- top = int(results['ymin'])
19
- left = int(results['xmin'])
20
- width = int(results['xmax'] - results['xmin'])
21
- height = int(results['ymax'] - results['ymin'])
22
  return {
23
  "top": top,
24
  "left": left,
 
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 {
23
  "top": top,
24
  "left": left,