panda1835 commited on
Commit
8d956c1
1 Parent(s): 97b72f1

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -69,7 +69,7 @@ class EndpointHandler():
69
  try:
70
  x1, y1, x2, y2 = result[0].boxes.xyxy.numpy().astype('int')[0]
71
  if result[0].boxes.conf[0].item() < 0.75: # if low in confidence
72
- x1, y1, x2, y2 = 0, 0, W, H
73
  else:
74
  annotated = result[0].plot(labels=False, conf=False)[:,:,::-1]
75
  except: # in case there is no detection
 
69
  try:
70
  x1, y1, x2, y2 = result[0].boxes.xyxy.numpy().astype('int')[0]
71
  if result[0].boxes.conf[0].item() < 0.75: # if low in confidence
72
+ return img.tolist(), "🤔 Hmm... Vích AI không thấy bạn rùa nào trong bức ảnh này. Bạn hãy tải lên một bức hình khác nhé."
73
  else:
74
  annotated = result[0].plot(labels=False, conf=False)[:,:,::-1]
75
  except: # in case there is no detection