Spaces:
Sleeping
Sleeping
Commit
·
fe3a33c
1
Parent(s):
4e4d6e2
Update app.py
Browse files
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-
|
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)
|