Update app.py

app.py

@@ -26,7 +26,21 @@ def inference(path:str, threshold:float=0.6):
                 color=(0, 0, 255),
                 thickness=2,
             )
-            cv2.putText(image, cls, (int(box[0]), int(box[2])))
+
+            font                   = cv2.FONT_HERSHEY_SIMPLEX
+            fontScale              = 1
+            fontColor              = (255,255,255)
+            thickness              = 1
+            lineType               = 2
+            
+            cv2.putText(img,str(cls), 
+            (int(box[0]), int(box[2])), 
+            font, 
+            fontScale,
+            fontColor,
+            thickness,
+            lineType)
+            
             counter+=1
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB), counter