Update handler.py
Browse files- 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 |
-
|
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
|