Update app.py

app.py

@@ -52,9 +52,9 @@ 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][0],"cat":1-predictions[0][0]}
-    
+        pdog=float(predictions[0][0])
+        return {"dog":pdog,"cat":1-pdog}
+    s
     except Exception as e:
         # Catch any error, log it, and return it to the user in a visible format
         error_message = f"CRITICAL RUNTIME ERROR: {str(e)}"