Dricz commited on
Commit
bb2ddcc
·
verified ·
1 Parent(s): 21c6654

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,8 +8,7 @@ model = YOLO('best (1).pt')
8
  def response(image):
9
  results = model(image)
10
  # Plot results image
11
- im_bgr = results.plot() # BGR-order numpy array
12
- im_rgb = im_bgr[..., ::-1] # Convert BGR to RGB
13
  return im_rgb
14
 
15
  iface = gr.Interface(fn=response, inputs="image", outputs="image")
 
8
  def response(image):
9
  results = model(image)
10
  # Plot results image
11
+
 
12
  return im_rgb
13
 
14
  iface = gr.Interface(fn=response, inputs="image", outputs="image")