Update app.py
Browse files
app.py
CHANGED
@@ -26,6 +26,7 @@ def inference(path:str, threshold:float=0.6):
|
|
26 |
color=(0, 0, 255),
|
27 |
thickness=2,
|
28 |
)
|
|
|
29 |
counter+=1
|
30 |
return cv2.cvtColor(image, cv2.COLOR_BGR2RGB), counter
|
31 |
|
|
|
26 |
color=(0, 0, 255),
|
27 |
thickness=2,
|
28 |
)
|
29 |
+
cv2.putText(image, cls, (int(box[0]), int(box[2])))
|
30 |
counter+=1
|
31 |
return cv2.cvtColor(image, cv2.COLOR_BGR2RGB), counter
|
32 |
|