Hugging Face's logo

Spaces:
abrar-adnan
/
cap-recognizer
Sleeping

App Files Community
abrar-adnan commited on
Commit
b852591
•
1 Parent(s): 5cb73ce

some update

Browse files
Files changed (1) hide show
  1. app.py +19 -19
app.py CHANGED
@@ -7,25 +7,25 @@ import gradio as gr
7
 
8
  cap_labels = (
9
  'balaclava cap',
10
- 'baseball cap',
11
- 'beanie cap',
12
- 'boater hat',
13
- 'bowler hat',
14
- 'bucket hat',
15
- 'cowboy hat',
16
- 'fedora cap',
17
- 'flat cap',
18
- 'ivy cap',
19
- 'kepi cap',
20
- 'newsboy cap',
21
- 'pork pie hat',
22
- 'rasta cap',
23
- 'sun hat',
24
- 'taqiyah cap',
25
- 'top hat',
26
- 'trucker cap',
27
- 'turban cap',
28
- 'visor cap'
29
  )
30
 
31
  model = load_learner('models/cap-recognizer-v1.pkl')
 
7
 
8
  cap_labels = (
9
  'balaclava cap',
10
+ 'baseball cap',
11
+ 'beanie cap',
12
+ 'boater hat',
13
+ 'bowler hat',
14
+ 'bucket hat',
15
+ 'cowboy hat',
16
+ 'fedora cap',
17
+ 'flat cap',
18
+ 'ivy cap',
19
+ 'kepi cap',
20
+ 'newsboy cap',
21
+ 'pork pie hat',
22
+ 'rasta cap',
23
+ 'sun hat',
24
+ 'taqiyah cap',
25
+ 'top hat',
26
+ 'trucker cap',
27
+ 'turban cap',
28
+ 'visor cap'
29
  )
30
 
31
  model = load_learner('models/cap-recognizer-v1.pkl')