achase25 commited on
Commit
30a588a
·
verified ·
1 Parent(s): 6904633

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -10,6 +10,7 @@ from transformers import AutoImageProcessor, AutoModelForImageClassification
10
 
11
  MODEL_ID = "valentinocc/dog-breed-classifier"
12
  AKC_CSV_PATH = "akc-data-latest.csv"
 
13
 
14
 
15
  # -----------Load model + processor-----------------------
@@ -75,6 +76,19 @@ ALIAS_DIRECT: Dict[str, str] = {
75
  "toy american eskimo dog": "American Eskimo Dog (Toy)",
76
  "miniature american eskimo dog": "American Eskimo Dog (Miniature)",
77
  "standard american eskimo dog": "American Eskimo Dog (Standard)",
 
 
 
 
 
 
 
 
 
 
 
 
 
78
  }
79
 
80
  # 2) Generic flip: "<Variant> <Base>" -> "<Base> (<Variant>)"
 
10
 
11
  MODEL_ID = "valentinocc/dog-breed-classifier"
12
  AKC_CSV_PATH = "akc-data-latest.csv"
13
+ DOG_LABELS_PATH = "dogmodelbreedlist.json"
14
 
15
 
16
  # -----------Load model + processor-----------------------
 
76
  "toy american eskimo dog": "American Eskimo Dog (Toy)",
77
  "miniature american eskimo dog": "American Eskimo Dog (Miniature)",
78
  "standard american eskimo dog": "American Eskimo Dog (Standard)",
79
+
80
+ # Others
81
+ "eskimo dog": "American Eskimo Dog",
82
+ "wire haired fox terrier": "Fox Terrier (Wire)",
83
+ "smooth fox terrier": "Fox Terrier (Smooth)",
84
+ "black and tan coonhound": "Black and Tan Coonhound",
85
+ "german short haired pointer": "German Shorthaired Pointer",
86
+ "german long haired pointer": "German Longhaired Pointer",
87
+ "curly coated retriever": "Curly-Coated Retriever",
88
+ "flat coated retriever": "Flat-Coated Retriever",
89
+ "yorkshire terrier": "Yorkshire Terrier",
90
+ "welsh springer spaniel": "Welsh Springer Spaniel",
91
+ "english springer": "English Springer Spaniel",
92
  }
93
 
94
  # 2) Generic flip: "<Variant> <Base>" -> "<Base> (<Variant>)"