Update app.py

app.py

@@ -23,7 +23,7 @@ def predict_dogs(image):
     probabilities = tf.nn.softmax(prediction)
     
     # Map probabilities to Pokemon classes
-    dogs_classes = ['Goldie', 'Beagle', 'Husky']  
+    dogs_classes = ['beagle', 'goldie', 'husky']  
     probabilities_dict = {dogs_class: round(float(probability), 2) for dogs_class, probability in zip(dogs_classes, probabilities[0])}
     
     return probabilities_dict
@@ -34,7 +34,6 @@ iface = gr.Interface(
     fn=predict_dogs,
     inputs=input_image, 
     outputs=gr.Label(),
-    live=True,
-    examples=["images/01.jpg", "images/02.jpg", "images/03.jpg", "images/04.jpg", "images/06.jpg", "images/06.jpg"],   
+    examples=["images/01.jpg", "images/02.jpg", "images/03.jpg", "images/04.jpg", "images/05.jpg", "images/06.jpg"],   
     description="A simple mlp classification model for image classification using the mnist dataset.")
 iface.launch()