Update app.py

app.py

@@ -314,13 +314,13 @@ async def predict(image):
         if len(dogs) <= 1:
             # Single dog or no dog detected, use the original method
             top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(image)
-            return process_single_dog_result(top1_prob, topk_breeds, topk_probs_percent, image)
+            return await process_single_dog_result(top1_prob, topk_breeds, topk_probs_percent, image)
         else:
             # Multiple dogs detected
-            return process_multiple_dogs_result(dogs, image)
+            return await process_multiple_dogs_result(dogs, image)
 
     except Exception as e:
-        return f"An error occurred: {e}", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return f"An error occurred: {str(e)}", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
 async def process_single_dog_result(top1_prob, topk_breeds, topk_probs_percent, image):
     if top1_prob >= 0.5:
@@ -391,7 +391,6 @@ async def show_details(choice):
     except Exception as e:
         return f"An error occurred while showing details: {e}"
 
-# Gradio 介面設置
 with gr.Blocks(css="""
     .container { max-width: 900px; margin: auto; padding: 20px; }
     .gr-box { border-radius: 15px; }
@@ -411,9 +410,17 @@ with gr.Blocks(css="""
     breed_buttons = gr.Radio([], label="Select breed for more details", visible=False)
     breed_details = gr.Markdown(label="Breed Details")
 
+    def handle_error(error):
+        return str(error), None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+
     input_image.change(
         predict,
         inputs=input_image,
+        outputs=[output, output_image, breed_buttons, breed_details],
+        api_name="predict"
+    ).catch(
+        fn=handle_error,
+        inputs=None,
         outputs=[output, output_image, breed_buttons, breed_details]
     )