Add reco for gradio
Browse files
app.py
CHANGED
@@ -143,7 +143,7 @@ def detect_logo(img, model, confidence):
|
|
143 |
f.write(('%g ' * (len(line)-1)).rstrip() % line[1:] + '\n')
|
144 |
|
145 |
if save_img or view_img:
|
146 |
-
label = f'{names[int(cls)]} {conf:.2f}
|
147 |
plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)
|
148 |
if view_img:
|
149 |
cv2.imshow(str(p), im0)
|
|
|
143 |
f.write(('%g ' * (len(line)-1)).rstrip() % line[1:] + '\n')
|
144 |
|
145 |
if save_img or view_img:
|
146 |
+
label = f'{names[int(cls)]} {conf:.2f} -> {name_recommanded.lower()}'
|
147 |
plot_one_box(xyxy, im0, label=label, color=colors[int(cls)], line_thickness=3)
|
148 |
if view_img:
|
149 |
cv2.imshow(str(p), im0)
|