Update app.py

app.py

@@ -53,7 +53,7 @@ def predict_image(input_img_pil):
         # 4. Format the output for Gradio's Label component
         # Assuming predictions is a 2-element array: [prob_cat, prob_dog]
         
-        return {"dog":predictions[0],"cat":1-predictions[0]}
+        return {"dog":predictions[0][0],"cat":1-predictions[0][0]}
     
     except Exception as e:
         # Catch any error, log it, and return it to the user in a visible format