DawnC commited on
Commit
82c1429
1 Parent(s): bbd78ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -10
app.py CHANGED
@@ -140,12 +140,8 @@ def preprocess_image(image):
140
  return transform(image).unsqueeze(0)
141
 
142
 
143
- def get_akc_link(breed):
144
- # Remove any non-English characters and convert to lowercase
145
- formatted_breed = ''.join(c for c in breed if ord(c) < 128).lower()
146
- # Replace spaces with hyphens and remove any remaining special characters
147
- formatted_breed = '-'.join(word for word in formatted_breed.split() if word.isalnum())
148
- return f"https://www.akc.org/dog-breeds/{formatted_breed}/"
149
 
150
  def predict(image):
151
  try:
@@ -160,7 +156,7 @@ def predict(image):
160
  breed = dog_breeds[predicted.item()]
161
 
162
  description = get_dog_description(breed)
163
- akc_link = get_akc_link(breed)
164
 
165
  if isinstance(description, dict):
166
  description_str = "\n\n".join([f"**{key}**: {value}" for key, value in description.items()])
@@ -168,11 +164,13 @@ def predict(image):
168
  description_str = description
169
 
170
  # Add AKC link as an option
171
- description_str += f"\n\n**Want to learn more?** [View detailed information about {breed} on the AKC website]({akc_link})"
172
 
173
  # Add disclaimer
174
- disclaimer = ("\n\n*Disclaimer: The external link provided leads to the American Kennel Club (AKC) website. "
175
- "We are not responsible for the content on external sites. Please refer to the AKC's terms of use and privacy policy.*")
 
 
176
  description_str += disclaimer
177
 
178
  return description_str
 
140
  return transform(image).unsqueeze(0)
141
 
142
 
143
+ def get_akc_breeds_link():
144
+ return "https://www.akc.org/dog-breeds/"
 
 
 
 
145
 
146
  def predict(image):
147
  try:
 
156
  breed = dog_breeds[predicted.item()]
157
 
158
  description = get_dog_description(breed)
159
+ akc_link = get_akc_breeds_link()
160
 
161
  if isinstance(description, dict):
162
  description_str = "\n\n".join([f"**{key}**: {value}" for key, value in description.items()])
 
164
  description_str = description
165
 
166
  # Add AKC link as an option
167
+ description_str += f"\n\n**Want to learn more about dog breeds?** [Visit the AKC dog breeds page]({akc_link}) and search for {breed} to find detailed information."
168
 
169
  # Add disclaimer
170
+ disclaimer = ("\n\n*Disclaimer: The external link provided leads to the American Kennel Club (AKC) dog breeds page. "
171
+ "You may need to search for the specific breed on that page. "
172
+ "I am not responsible for the content on external sites. "
173
+ "Please refer to the AKC's terms of use and privacy policy.*")
174
  description_str += disclaimer
175
 
176
  return description_str