app updated

app.py

@@ -27,7 +27,7 @@ def predict(img):
 
     # Create a prediction label and prediction probability dictionary for each prediction class
     # This is the required format for Gradio's output parameter
-    pred_labels_and_probs = {'dog' if MODEL.predict(features)> 0.5 else 'cat':float(MODEL.predict(features))}
+    pred_labels_and_probs = {'dog':float(MODEL.predict(features))} if MODEL.predict(features)> 0.5 else {'cat':100-float(MODEL.predict(features))}
 
     # Calculate the prediction time
     pred_time = round(timer() - start_time, 5)