Akshat-1812 commited on
Commit
12aa0f9
·
1 Parent(s): 69363a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -36
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
- 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):
 
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):