Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,16 @@ import gradio as gr
|
|
13 |
learn_inf = load_learner('export2.pkl')
|
14 |
|
15 |
# %% ../dog-breeds.ipynb 15
|
16 |
-
categories =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
def classify_image(img):
|
19 |
pred, idx, probs = learn_inf.predict(img)
|
|
|
13 |
learn_inf = load_learner('export2.pkl')
|
14 |
|
15 |
# %% ../dog-breeds.ipynb 15
|
16 |
+
categories = ['Abyssinian', 'Bengal', 'Birman', 'Bombay', 'British_Shorthair',
|
17 |
+
'Egyptian_Mau', 'Maine_Coon', 'Persian', 'Ragdoll', 'Russian_Blue',
|
18 |
+
'Siamese', 'Sphynx', 'american_bulldog', 'american_pit_bull_terrier',
|
19 |
+
'basset_hound', 'beagle', 'boxer', 'chihuahua', 'english_cocker_spaniel',
|
20 |
+
'english_setter', 'german_shorthaired', 'great_pyrenees', 'havanese',
|
21 |
+
'japanese_chin', 'keeshond', 'leonberger', 'miniature_pinscher',
|
22 |
+
'newfoundland', 'pomeranian', 'pug', 'saint_bernard', 'samoyed',
|
23 |
+
'scottish_terrier', 'shiba_inu', 'staffordshire_bull_terrier',
|
24 |
+
'wheaten_terrier', 'yorkshire_terrier']
|
25 |
+
|
26 |
|
27 |
def classify_image(img):
|
28 |
pred, idx, probs = learn_inf.predict(img)
|