Update app.py

app.py

@@ -8,7 +8,6 @@ import torch.nn.functional as F
 from torchvision import transforms
 from PIL import Image
 from data_manager import get_dog_description
-import wikipedia
 from urllib.parse import quote
 
 dog_breeds = ["Afghan_Hound(阿富汗獵犬)", "African_Hunting_Dog(非洲野犬)", "Airedale(艾爾谷犬)",
@@ -187,10 +186,7 @@ def predict(image):
         description = get_dog_description(breed)
         
         # Generate Wikipedia link
-        try:
-            wiki_link = wikipedia.page(f"{breed} dog").url
-        except:
-            wiki_link = f"https://en.wikipedia.org/wiki/Special:Search?search={quote(breed)}+dog"
+        wiki_link = f"https://en.wikipedia.org/wiki/Special:Search?search={quote(breed)}+dog"
         
         # Formatting the description for better display
         if isinstance(description, dict):
@@ -207,7 +203,7 @@ def predict(image):
             description_str = description
         
         # Add Wikipedia link
-        description_str += f"\n\n[Click here to view the Wikipedia page for {breed}]({wiki_link})"
+        description_str += f"\n\n[Click here to search for {breed} on Wikipedia]({wiki_link})"
         
         return description_str
     except Exception as e: