Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ def predict_image(image):
|
|
| 61 |
results = model(temp_path, imgsz=640, conf=0.3)
|
| 62 |
result = results[0]
|
| 63 |
|
| 64 |
-
pred_img = result.plot()
|
| 65 |
|
| 66 |
boxes = result.boxes.data.cpu().numpy()
|
| 67 |
cls_names = result.names
|
|
|
|
| 61 |
results = model(temp_path, imgsz=640, conf=0.3)
|
| 62 |
result = results[0]
|
| 63 |
|
| 64 |
+
pred_img = result.plot(conf=False)
|
| 65 |
|
| 66 |
boxes = result.boxes.data.cpu().numpy()
|
| 67 |
cls_names = result.names
|