princeml commited on
Commit
b0f5f2e
·
1 Parent(s): e3580eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model = YOLO("best.pt") # load a custom model
19
  def gradio_wrapper(img):
20
  global model
21
  #print(np.shape(img))
22
- results = model(img, save = True, show = True) # predict on an image
23
  # print(model.names.get(box.cls.item()))
24
 
25
  # return cv2.putText(img, str(results[0]),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1,
 
19
  def gradio_wrapper(img):
20
  global model
21
  #print(np.shape(img))
22
+ results = model(img, show = True) # predict on an image
23
  # print(model.names.get(box.cls.item()))
24
 
25
  # return cv2.putText(img, str(results[0]),(00, 185), cv2.FONT_HERSHEY_SIMPLEX, 1,