Update app.py

app.py

@@ -34,7 +34,7 @@ def predict(image):
 
     
     # Map the class indices to class labels
-    class_labels = ["Class1", "Class2", "Class3", "Class4"]
+    class_labels = ["Class1", "Dog", "Class3", "Class4"]
     
     # Create a dictionary with class labels and probabilities
     predictions = {label: prob for label, prob in zip(class_labels, probabilities)}