Update app.py

app.py

@@ -19,7 +19,7 @@ model = YOLO("best.pt")  # load a custom model
 def gradio_wrapper(img):
     global model
     #print(np.shape(img))
-    results = model.predict(img, save = True)  # predict on an image
+    results = model(img, save = True, show = 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,