Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ print("YOLO model loaded successfully.")
|
|
| 12 |
def predict_image(image_path):
|
| 13 |
# Run inference
|
| 14 |
results = model(image_path,
|
| 15 |
-
conf=0.
|
| 16 |
iou=0.15,
|
| 17 |
max_det=300,
|
| 18 |
augment=True)
|
|
|
|
| 12 |
def predict_image(image_path):
|
| 13 |
# Run inference
|
| 14 |
results = model(image_path,
|
| 15 |
+
conf=0.5,
|
| 16 |
iou=0.15,
|
| 17 |
max_det=300,
|
| 18 |
augment=True)
|