Update app.py

app.py

@@ -39,7 +39,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 = 'dog' if predicted.item() ==1  else 'cat'
+    pred_labels = 'Cat' if predicted.item() ==1  else 'Dog'
 
     # Calculate the prediction time
     pred_time = round(timer() - start_time, 5)