Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,10 @@ from transformers import pipeline
|
|
9 |
# bumblebee, honeybee, vespidae,
|
10 |
# then 0 => bumblebee, 1 => honeybee, 2 => vespidae (the default PyTorch order).
|
11 |
# Verify your label indices by printing `test_dataset.classes` in your training script.
|
|
|
12 |
label_map = {
|
13 |
-
"LABEL_0": "
|
14 |
-
"LABEL_1": "
|
15 |
"LABEL_2": "vespidae"
|
16 |
}
|
17 |
|
|
|
9 |
# bumblebee, honeybee, vespidae,
|
10 |
# then 0 => bumblebee, 1 => honeybee, 2 => vespidae (the default PyTorch order).
|
11 |
# Verify your label indices by printing `test_dataset.classes` in your training script.
|
12 |
+
|
13 |
label_map = {
|
14 |
+
"LABEL_0": "bumblebee",
|
15 |
+
"LABEL_1": "honeybee",
|
16 |
"LABEL_2": "vespidae"
|
17 |
}
|
18 |
|