Update app.py

app.py

@@ -20,11 +20,11 @@ def gradio_wrapper(img):
     global model
     #print(np.shape(img))
     results = model.predict(img)  # predict on an image
-    # print(model.names.get(box.cls.item()))
+    print(results[0].boxes.cls)
     
-    # return cv2.putText(img, str(results[0]),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
-    #               (0, 0, 255),  2, cv2.LINE_AA, False)
-    return results
+    return cv2.putText(img, str(results[0].boxes.cls),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
+                  (0, 0, 255),  2, cv2.LINE_AA, False)
+    # return results
   
 demo = gr.Interface(
     gradio_wrapper,