Hugging Face's logo

Spaces:
006f86
/
StrAI
Sleeping

App Files Community
006f86 commited on
Commit
cdb44c0
·
1 Parent(s): b4288d3

added classes list (references JSON instead of manually adding classes in app.py)

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. class_names.json +161 -0
app.py CHANGED
@@ -8,7 +8,8 @@ import json
8
  import os
9
 
10
  MODEL_PATH = "best_model.pth"
11
- CLASS_NAMES = ["Achillies", "Adolf", "Al Waldo", "Alfie", "Ally", "Anselmo", "Baby Belgian", "Balls", "Bambi", "Bangus", "Barry", "Bingo", "Bingus", "Borgee", "Bozo", "Bulgogi", "Butter", "Butter", "Button", "Callie", "Cara", "Carme", "Chichi Shopwise", "Chico", "Choweder", "Cleo", "Coconut", "Daisy", "Daphne", "Darlene", "DOG 2", "DOG 3", "Dorian", "Doris", "Dot", "Ebi", "Echo", "Elise", "Enji", "Ensaymada Cheese Roll", "Faith", "Fave", "Flora", "Floyd", "Francis", "Frankie", "Georgie", "Gigi", "Ginger", "Gruyere", "Hansel", "Hany", "Harith", "Hazel", "Helga", "Huey", "Huni", "Inu", "Jack", "Jackie", "Jerome", "Josie", "JR", "Juliet", "Julio", "Juno", "Kalbi", "Kenta", "Kimchi", "Kimono", "KittyPerry", "Kucing", "Kucingcing", "Lambing", "Lara", "Leeca", "Lennon", "Leslie", "Limper_Bulag", "Limpkin", "Lion", "Louis", "Lucky", "Lyka", "Maggie", "Mama Waffle", "Maple Oscar", "Marge", "Marikit", "Marty", "Maxie", "Meemon", "Meowming", "Miki", "Milo", "Mingkay", "Minty", "Mitzki", "Moji", "Munchie", "Nala", "Nemo", "Nero", "NO_NAME 3", "NO_NAME 4", "NO_NAME 5", "NO_NAME 6", "NO_NAME 7- Batgirl", "NO_NAME 11", "NO_NAME 13 - Jinu", "NO_NAME Security Cat", "not_faith", "Nyssa Nisa", "Paris", "Patches", "Patchot", "Peeta", "Penny", "Pepper", "Percy", "Prof", "Pumpkin", "Pumpkin", "Raj", "Remy", "Renz", "Ringo", "Rocky", "Rory", "Ross", "Salvi", "Sam", "Shaq", "Sheeba", "Shibuya", "Sniper", "Socks", "Spaghetti", "Sunkiss", "Suzy", "Tarsier", "Tasi", "Teddy Chi", "Toes", "Toffee", "Toothless", "Trex", "Tutti", "Twister Fries", "Val", "Void", "Waffle", "Weena"]
 
12
 
13
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
14
 
 
8
  import os
9
 
10
  MODEL_PATH = "best_model.pth"
11
+ with open("class_names.json", "r") as f:
12
+ CLASS_NAMES = json.load(f)
13
 
14
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
15
 
class_names.json ADDED
@@ -0,0 +1,161 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ "Achillies",
3
+ "Adolf",
4
+ "Al Waldo",
5
+ "Alfie",
6
+ "Ally",
7
+ "Anselmo",
8
+ "Baby Belgian",
9
+ "Balls",
10
+ "Bambi",
11
+ "Bangus",
12
+ "Barry",
13
+ "Bingo",
14
+ "Bingus",
15
+ "Borgee",
16
+ "Bozo",
17
+ "Bulgogi",
18
+ "Butter",
19
+ "Button",
20
+ "Callie",
21
+ "Cara",
22
+ "Carme",
23
+ "Chichi Shopwise",
24
+ "Chico",
25
+ "Chowder",
26
+ "Cleo",
27
+ "Coconut",
28
+ "DOG 2",
29
+ "DOG 3",
30
+ "Daisy",
31
+ "Daphne",
32
+ "Darlene",
33
+ "Dorian",
34
+ "Doris",
35
+ "Dot",
36
+ "Ebi",
37
+ "Echo",
38
+ "Elise",
39
+ "Enji",
40
+ "Ensaymada Cheese Roll",
41
+ "Faith",
42
+ "Fave",
43
+ "Flora",
44
+ "Floyd",
45
+ "Francis",
46
+ "Frankie",
47
+ "Georgie",
48
+ "Gigi",
49
+ "Ginger",
50
+ "Gruyere",
51
+ "Hansel",
52
+ "Hany",
53
+ "Harith",
54
+ "Hazel",
55
+ "Helga",
56
+ "Huey",
57
+ "Huni",
58
+ "Inu",
59
+ "JR",
60
+ "Jack",
61
+ "Jackie",
62
+ "Jerome",
63
+ "Josie",
64
+ "Juliet",
65
+ "Julio",
66
+ "Juno",
67
+ "Kalbi",
68
+ "Kenta",
69
+ "Kimchi",
70
+ "Kimono",
71
+ "KittyPerry",
72
+ "Kucing",
73
+ "Kucingcing",
74
+ "Lambing",
75
+ "Lara",
76
+ "Leeca",
77
+ "Lennon",
78
+ "Leslie",
79
+ "Lily",
80
+ "Limper_Bulag",
81
+ "Limpkin",
82
+ "Lion",
83
+ "Louis",
84
+ "Lucky",
85
+ "Lyka",
86
+ "Maggie",
87
+ "Mama Waffle",
88
+ "Maple Oscar",
89
+ "Marga",
90
+ "Marikit",
91
+ "Marty",
92
+ "Maxie",
93
+ "Meemon",
94
+ "Meowming",
95
+ "Miki",
96
+ "Milo",
97
+ "Mingkay",
98
+ "Minty",
99
+ "Mitzki",
100
+ "Moji",
101
+ "Munchie",
102
+ "NO_NAME 11",
103
+ "NO_NAME 13 - Jinu",
104
+ "NO_NAME 3",
105
+ "NO_NAME 4",
106
+ "NO_NAME 5",
107
+ "NO_NAME 6",
108
+ "NO_NAME 7 - Batgirl",
109
+ "NO_NAME Security Cat",
110
+ "Nala",
111
+ "Nemo",
112
+ "Nero",
113
+ "Nyssa Nisa",
114
+ "Paris",
115
+ "Patches",
116
+ "Patchot",
117
+ "Peeta",
118
+ "Penny",
119
+ "Pepper",
120
+ "Percy",
121
+ "Prof",
122
+ "Pumpkin",
123
+ "Raj",
124
+ "Remy",
125
+ "Renz",
126
+ "Ringo",
127
+ "Rocky",
128
+ "Rory",
129
+ "Ross",
130
+ "Salvi",
131
+ "Sam",
132
+ "Shaq",
133
+ "Sheeba",
134
+ "Shibuya",
135
+ "Sniper",
136
+ "Socks",
137
+ "Soup",
138
+ "Spaghetti",
139
+ "Sunkiss",
140
+ "Suzy",
141
+ "Tarsier",
142
+ "Tasi",
143
+ "Teddy Chi",
144
+ "Toes",
145
+ "Toffee",
146
+ "Toothless",
147
+ "Trex",
148
+ "Tutti",
149
+ "Twister Fries",
150
+ "Val",
151
+ "Void",
152
+ "Waffle",
153
+ "Weena",
154
+ "Whiskers",
155
+ "Wilma",
156
+ "Yoyo",
157
+ "Yuris",
158
+ "Ziggy",
159
+ "Zorro",
160
+ "not_faith"
161
+ ]