halictus commited on
Commit
53e6e66
·
verified ·
1 Parent(s): b6e309c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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": "honeybee",
14
- "LABEL_1": "bumblebee",
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