Update app.py
Browse files
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 |
-
|
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")
|