Update app.py

app.py

@@ -21,7 +21,7 @@ def predict_pokemon(image):
     probabilities = tf.nn.softmax(prediction, axis=1)
     
     # Map probabilities to Pokemon classes
-    pokemon_classes = ['Articuno', 'Bulbasaur', 'Charmander']  
+    class_names = ['Chansey', 'Growlithe', 'Lapras']
     probabilities_dict = {pokemon_class: round(float(probability), 2) for pokemon_class, probability in zip(pokemon_classes, probabilities.numpy()[0])}
     
     return probabilities_dict
@@ -45,6 +45,6 @@ if uploaded_image is not None:
 
 # Example images
 st.sidebar.title("Examples")
-example_images = ["images/01.jpg", "images/02.png", "images/03.png", "images/04.jpg", "images/05.png", "images/06.png"]
+example_images = ["pokemon/train/chansey/00000000.png", "pokemon/train/growlithe/00000000.png", "pokemon/train/lapras/00000000.png"]
 for example_image in example_images:
     st.sidebar.image(example_image, use_column_width=True)