mmchowdhury commited on
Commit
fe3a33c
·
1 Parent(s): 4e4d6e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -4,6 +4,8 @@ import gradio as gr
4
 
5
  labels=(
6
  'Abyssinian cat',
 
 
7
  'Bengal cat',
8
  'Birman cat',
9
  'Bombay cat',
@@ -12,9 +14,17 @@ labels=(
12
  'Burmilla cat',
13
  'Cornish Rex cat',
14
  'Cymric cat',
15
- 'Donskoy cat'
 
 
 
 
 
 
 
 
16
  )
17
- model=load_learner(f'CatClassifier-v0.pkl')
18
  def recognize_image(image):
19
  pred, idx, probs=model.predict(image)
20
  print(pred)
 
4
 
5
  labels=(
6
  'Abyssinian cat',
7
+ 'Aegean cat',
8
+ 'Balinese cat',
9
  'Bengal cat',
10
  'Birman cat',
11
  'Bombay cat',
 
14
  'Burmilla cat',
15
  'Cornish Rex cat',
16
  'Cymric cat',
17
+ 'Donskoy cat',
18
+ 'Oregon Rex cat',
19
+ 'Oriental Bicolor cat',
20
+ 'Persian cat',
21
+ 'Pixie-Bob cat',
22
+ 'Ragamuffin cat',
23
+ 'Siamese cat',
24
+ 'Siberian cat',
25
+ 'Turkish Angora cat'
26
  )
27
+ model=load_learner(f'CatClassifier-v1.pkl')
28
  def recognize_image(image):
29
  pred, idx, probs=model.predict(image)
30
  print(pred)