Modified predicted_label

app.py

@@ -13,13 +13,8 @@ def predict_image(image):
     # Make prediction
     pred = model.predict(image)
 
-    predicted_label = ""
-    if pred >= 0.5:
-        predicted_label = " Cat"
-    else:
-        predicted_label = "Dog"
-
-    predicted_label = f"{predicted_label} with prediction : {pred[0][0]}"
+    predicted_label = f"{(1-pred[0][0])*100}% Dog \n{pred[0][0]*100}% Cat \nFinal Prediction : {'Dog' if pred[0][0] < 0.5 else 'Cat'}"
+    
     # Return the predicted label
     return predicted_label