bishmoy commited on
Commit
e47a3dd
·
1 Parent(s): 8d4307c

Fixed small issue

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -6,14 +6,14 @@ import gradio as gr
6
  IMAGE_SIZE = 72
7
  # labels taken from https://huggingface.co/datasets/cifar10
8
  labels = {0: "airplane",
9
- 1: "automobile"
10
- 2: "bird"
11
- 3: "cat"
12
- 4: "deer"
13
- 5: "dog"
14
- 6: "frog"
15
- 7: "horse"
16
- 8: "ship"
17
  9: "truck"}
18
 
19
 
 
6
  IMAGE_SIZE = 72
7
  # labels taken from https://huggingface.co/datasets/cifar10
8
  labels = {0: "airplane",
9
+ 1: "automobile",
10
+ 2: "bird",
11
+ 3: "cat",
12
+ 4: "deer",
13
+ 5: "dog",
14
+ 6: "frog",
15
+ 7: "horse",
16
+ 8: "ship",
17
  9: "truck"}
18
 
19