Update app.py

app.py

@@ -24,7 +24,7 @@ def predict_pokemon(image):
     probabilities = tf.nn.softmax(prediction, axis=1)
     
     # Map probabilities to Pokemon classes
-    class_names = ['Chansey', 'Growlithe', 'Lapras']
+    class_names = ['Charizard', 'Lapras', 'Machamp']
     probabilities_dict = {pokemon_class: round(float(probability), 2) for pokemon_class, probability in zip(class_names, probabilities.numpy()[0])}
     
     return probabilities_dict
@@ -47,19 +47,4 @@ if uploaded_image is not None:
     # Display predictions as a DataFrame
     st.write("### Prediction Probabilities")
     df = pd.DataFrame(predictions.items(), columns=["Pokemon", "Probability"])
-    st.dataframe(df)
-    
-    # Display predictions as a bar chart
-    st.write("### Prediction Chart")
-    fig, ax = plt.subplots()
-    ax.barh(df["Pokemon"], df["Probability"], color='skyblue')
-    ax.set_xlim(0, 1)
-    ax.set_xlabel('Probability')
-    ax.set_title('Prediction Probabilities')
-    st.pyplot(fig)
-
-# Example images
-st.sidebar.title("Examples")
-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)
+    st.dataframe(df)