Update app.py

app.py

@@ -19,11 +19,12 @@ model = YOLO("best.pt")  # load a custom model
 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()))
+    results = model.predict(img, save = True)  # predict on an image
+    # print(model.names.get(box.cls.item()))
     
-    return cv2.putText(img, str(results[0]),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
-                  (0, 0, 255),  2, cv2.LINE_AA, False)
+    # return cv2.putText(img, str(results[0]),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1, 
+    #               (0, 0, 255),  2, cv2.LINE_AA, False)
+    return results
   
 demo = gr.Interface(
     gradio_wrapper,