Update app.py

app.py

@@ -194,16 +194,28 @@ def predict(image):
             # 檢查最高的預測機率
             top1_prob = topk_probs[0][0].item()
 
-            if top1_prob >= 0.5:
+            if top1_prob >= 0.6:
                 # 正確辨識時，返回該品種資訊
                 predicted = topk_indices[0][0]
                 breed = dog_breeds[predicted.item()]
                 description = get_dog_description(breed)
                 akc_link = get_akc_breeds_link()
 
-                description_str = f"**Breed**: {breed}\n\n**Description**: {description}"
+                if isinstance(description, dict):
+                    description_str = "\n\n".join([f"**{key}**: {value}" for key, value in description.items()])
+                else:
+                    description_str = description
+
+                # 添加AKC連結
                 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."
 
+                # 添加免責聲明
+                disclaimer = ("\n\n*Disclaimer: The external link provided leads to the American Kennel Club (AKC) dog breeds page. "
+                              "You may need to search for the specific breed on that page. "
+                              "I am not responsible for the content on external sites. "
+                              "Please refer to the AKC's terms of use and privacy policy.*")
+                description_str += disclaimer
+
                 return description_str
 
             else:
@@ -226,7 +238,6 @@ def predict(image):
     except Exception as e:
         return f"An error occurred: {e}"
 
-
 iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(label="Upload a dog image", type="numpy"),