Spaces:
Runtime error
Runtime error
Commit
·
12aa0f9
1
Parent(s):
69363a4
Update app.py
Browse files
app.py
CHANGED
@@ -6,42 +6,42 @@ import tensorflow_hub as hub
|
|
6 |
path = '20220804-16551659632113-all-images-Adam.h5'
|
7 |
model = tf.keras.models.load_model(path,custom_objects={"KerasLayer":hub.KerasLayer})
|
8 |
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
|
46 |
# load the model
|
47 |
def predict_breed(image):
|
|
|
6 |
path = '20220804-16551659632113-all-images-Adam.h5'
|
7 |
model = tf.keras.models.load_model(path,custom_objects={"KerasLayer":hub.KerasLayer})
|
8 |
|
9 |
+
labels = ['affenpinscher', 'afghan_hound', 'african_hunting_dog', 'airedale',
|
10 |
+
'american_staffordshire_terrier', 'appenzeller',
|
11 |
+
'australian_terrier', 'basenji', 'basset', 'beagle',
|
12 |
+
'bedlington_terrier', 'bernese_mountain_dog',
|
13 |
+
'black-and-tan_coonhound', 'blenheim_spaniel', 'bloodhound',
|
14 |
+
'bluetick', 'border_collie', 'border_terrier', 'borzoi',
|
15 |
+
'boston_bull', 'bouvier_des_flandres', 'boxer',
|
16 |
+
'brabancon_griffon', 'briard', 'brittany_spaniel', 'bull_mastiff',
|
17 |
+
'cairn', 'cardigan', 'chesapeake_bay_retriever', 'chihuahua',
|
18 |
+
'chow', 'clumber', 'cocker_spaniel', 'collie',
|
19 |
+
'curly-coated_retriever', 'dandie_dinmont', 'dhole', 'dingo',
|
20 |
+
'doberman', 'english_foxhound', 'english_setter',
|
21 |
+
'english_springer', 'entlebucher', 'eskimo_dog',
|
22 |
+
'flat-coated_retriever', 'french_bulldog', 'german_shepherd',
|
23 |
+
'german_short-haired_pointer', 'giant_schnauzer',
|
24 |
+
'golden_retriever', 'gordon_setter', 'great_dane',
|
25 |
+
'great_pyrenees', 'greater_swiss_mountain_dog', 'groenendael',
|
26 |
+
'ibizan_hound', 'irish_setter', 'irish_terrier',
|
27 |
+
'irish_water_spaniel', 'irish_wolfhound', 'italian_greyhound',
|
28 |
+
'japanese_spaniel', 'keeshond', 'kelpie', 'kerry_blue_terrier',
|
29 |
+
'komondor', 'kuvasz', 'labrador_retriever', 'lakeland_terrier',
|
30 |
+
'leonberg', 'lhasa', 'malamute', 'malinois', 'maltese_dog',
|
31 |
+
'mexican_hairless', 'miniature_pinscher', 'miniature_poodle',
|
32 |
+
'miniature_schnauzer', 'newfoundland', 'norfolk_terrier',
|
33 |
+
'norwegian_elkhound', 'norwich_terrier', 'old_english_sheepdog',
|
34 |
+
'otterhound', 'papillon', 'pekinese', 'pembroke', 'pomeranian',
|
35 |
+
'pug', 'redbone', 'rhodesian_ridgeback', 'rottweiler',
|
36 |
+
'saint_bernard', 'saluki', 'samoyed', 'schipperke',
|
37 |
+
'scotch_terrier', 'scottish_deerhound', 'sealyham_terrier',
|
38 |
+
'shetland_sheepdog', 'shih-tzu', 'siberian_husky', 'silky_terrier',
|
39 |
+
'soft-coated_wheaten_terrier', 'staffordshire_bullterrier',
|
40 |
+
'standard_poodle', 'standard_schnauzer', 'sussex_spaniel',
|
41 |
+
'tibetan_mastiff', 'tibetan_terrier', 'toy_poodle', 'toy_terrier',
|
42 |
+
'vizsla', 'walker_hound', 'weimaraner', 'welsh_springer_spaniel',
|
43 |
+
'west_highland_white_terrier', 'whippet',
|
44 |
+
'wire-haired_fox_terrier', 'yorkshire_terrier']
|
45 |
|
46 |
# load the model
|
47 |
def predict_breed(image):
|