Update app.py
Browse files
app.py
CHANGED
@@ -10,9 +10,9 @@ model = YOLO('best.pt')
|
|
10 |
def normalize_preds(label):
|
11 |
|
12 |
if label =="Blank":
|
13 |
-
|
14 |
|
15 |
-
return
|
16 |
|
17 |
def show_preds_image(image):
|
18 |
# Save the uploaded image temporarily
|
|
|
10 |
def normalize_preds(label):
|
11 |
|
12 |
if label =="Blank":
|
13 |
+
label = "Deer"
|
14 |
|
15 |
+
return label
|
16 |
|
17 |
def show_preds_image(image):
|
18 |
# Save the uploaded image temporarily
|