Update app.py

app.py

@@ -410,17 +410,15 @@ 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)
+    async def safe_predict(image):
+        try:
+            return await predict(image)
+        except Exception as e:
+            return str(e), None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
     input_image.change(
-        predict,
+        safe_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]
     )