Update app.py

app.py

@@ -10,55 +10,6 @@ from PIL import Image
 from data_manager import get_dog_description
 from urllib.parse import quote
 
-# dog_breeds = ["Afghan_Hound(阿富汗獵犬)", "African_Hunting_Dog(非洲野犬)", "Airedale(艾爾谷犬)",
-#             "American_Staffordshire_Terrier(美國斯塔福郡梗)", "Appenzeller(亞賓澤爾犬)",
-#             "Australian_Terrier(澳大利亞梗)", "Bedlington_Terrier(貝德靈頓梗)",
-#             "Bernese_Mountain_Dog(伯恩山犬)", "Blenheim_Spaniel(布萊尼姆獵犬)",
-#             "Border_Collie(邊境牧羊犬)", "Border_Terrier(邊境梗)", "Boston_Bull(波士頓梗)",
-#             "Bouvier_Des_Flandres(法蘭德斯牧羊犬)", "Brabancon_Griffon(布魯塞爾格里芬犬)",
-#             "Brittany_Spaniel(布列塔尼獵犬)", "Cardigan(卡迪根威爾士柯基犬)",
-#             "Chesapeake_Bay_Retriever(切薩皮克灣獵犬)", "Chihuahua(吉娃娃)",
-#             "Dandie_Dinmont(丹第丁蒙梗)", "Doberman(杜賓犬)", "English_Foxhound(英國獵狐犬)",
-#             "English_Setter(英國雪達犬)", "English_Springer(英國跳獵犬)",
-#             "EntleBucher(恩特雷布赫山地犬)", "Eskimo_Dog(愛斯基摩犬)", "French_Bulldog(法國鬥牛犬)",
-#             "German_Shepherd(德國牧羊犬)", "German_Short-Haired_Pointer(德國短毛指示犬)",
-#             "Gordon_Setter(戈登雪達犬)", "Great_Dane(大丹犬)", "Great_Pyrenees(大白熊犬)",
-#             "Greater_Swiss_Mountain_Dog(大瑞士山地犬)", "Ibizan_Hound(依比沙獵犬)",
-#             "Irish_Setter(愛爾蘭雪達犬)", "Irish_Terrier(愛爾蘭梗)",
-#             "Irish_Water_Spaniel(愛爾蘭水獵犬)", "Irish_Wolfhound(愛爾蘭獵狼犬)",
-#             "Italian_Greyhound(義大利灰狗)", "Japanese_Spaniel(日本狆)",
-#             "Kerry_Blue_Terrier(凱利藍梗)", "Labrador_Retriever(拉布拉多尋回犬)",
-#             "Lakeland_Terrier(湖畔梗)", "Leonberg(獅毛狗)", "Lhasa(拉薩犬)",
-#             "Maltese_Dog(馬爾濟斯犬)", "Mexican_Hairless(墨西哥無毛犬)", "Newfoundland(紐芬蘭犬)",
-#             "Norfolk_Terrier(諾福克梗)", "Norwegian_Elkhound(挪威獵麋犬)",
-#             "Norwich_Terrier(諾利治梗)", "Old_English_Sheepdog(古代英國牧羊犬)",
-#             "Pekinese(北京犬)", "Pembroke(威爾士柯基犬)", "Pomeranian(博美犬)",
-#             "Rhodesian_Ridgeback(羅得西亞脊背犬)", "Rottweiler(羅威納犬)",
-#             "Saint_Bernard(聖伯納犬)", "Saluki(薩路基獵犬)", "Samoyed(薩摩耶犬)",
-#             "Scotch_Terrier(蘇格蘭梗)", "Scottish_Deerhound(蘇格蘭獵鹿犬)",
-#             "Sealyham_Terrier(錫利哈姆梗)", "Shetland_Sheepdog(設得蘭牧羊犬)",
-#             "Shih-Tzu(西施犬)", "Siberian_Husky(西伯利亞哈士奇)",
-#             "Staffordshire_Bullterrier(斯塔福郡鬥牛梗)", "Sussex_Spaniel(蘇塞克斯獵犬)",
-#             "Tibetan_Mastiff(藏獒)", "Tibetan_Terrier(西藏梗)", "Walker_Hound(沃克獵犬)",
-#             "Weimaraner(威瑪犬)", "Welsh_Springer_Spaniel(威爾士跳獵犬)",
-#             "West_Highland_White_Terrier(西高地白梗)", "Yorkshire_Terrier(約克夏梗)",
-#             "Affenpinscher(猴犬)", "Basenji(巴辛吉犬)", "Basset(巴吉度獵犬)", "Beagle(比格犬)",
-#             "Black-and-Tan_Coonhound(黑褐獵浣熊犬)", "Bloodhound(尋血獵犬)",
-#             "Bluetick(布魯提克獵犬)", "Borzoi(俄羅斯獵狼犬)", "Boxer(拳師犬)", "Briard(布里亞犬)",
-#             "Bull_Mastiff(獒犬)", "Cairn(凱恩梗)", "Chow(鬆獅犬)", "Clumber(克倫伯獵犬)",
-#             "Cocker_Spaniel(可卡獵犬)", "Collie(柯利牧羊犬)", "Curly-Coated_Retriever(捲毛尋回犬)",
-#             "Dhole(豺)", "Dingo(澳洲野犬)", "Flat-Coated_Retriever(平毛尋回犬)",
-#             "Giant_Schnauzer(大型雪納瑞犬)", "Golden_Retriever(黃金獵犬)",
-#             "Groenendael(比利時牧羊犬)", "Keeshond(荷蘭毛獅犬)", "Kelpie(澳洲卡爾比犬)",
-#             "Komondor(匈牙利牧羊犬)", "Kuvasz(庫瓦茲犬)", "Malamute(阿拉斯加雪橇犬)",
-#             "Malinois(比利時瑪利諾犬)", "Miniature_Pinscher(迷你杜賓犬)",
-#             "Miniature_Poodle(迷你貴賓犬)", "Miniature_Schnauzer(迷你雪納瑞犬)",
-#             "Otterhound(水獺獵犬)", "Papillon(蝴蝶犬)", "Pug(巴哥犬)", "Redbone(紅骨獵浣熊犬)",
-#             "Schipperke(舒柏奇犬)", "Silky_Terrier(絲毛梗)",
-#             "Soft-Coated_Wheaten_Terrier(愛爾蘭軟毛梗)", "Standard_Poodle(標準貴賓犬)",
-#             "Standard_Schnauzer(標準雪納瑞犬)", "Toy_Poodle(玩具貴賓犬)", "Toy_Terrier(玩具梗)",
-#             "Vizsla(維茲拉犬)", "Whippet(惠比特犬)", "Wire-Haired_Fox_Terrier(硬毛獵狐梗)"]
-
 dog_breeds = ["Afghan_Hound", "African_Hunting_Dog", "Airedale", "American_Staffordshire_Terrier", 
               "Appenzeller", "Australian_Terrier", "Bedlington_Terrier", "Bernese_Mountain_Dog", 
               "Blenheim_Spaniel", "Border_Collie", "Border_Terrier", "Boston_Bull", "Bouvier_Des_Flandres", 
@@ -192,7 +143,7 @@ def predict(image):
                 description = get_dog_description(breed)
                 return format_description(description, breed), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
-            elif top1_prob < 0.1:
+            elif top1_prob < 0.2:
                 return ("The image is too unclear or the dog breed is not in the dataset. Please upload a clearer image of the dog.",
                         gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
             else: