Update app.py

app.py

@@ -43,10 +43,15 @@ if uploaded_image is not None:
     st.write("Classifying...")
     
     predictions = predict_pokemon(image)
+
+    # Display predictions as a DataFrame
+    st.write("### Prediction Probabilities")
+    df = pd.DataFrame(predictions.items(), columns=["Pokemon", "Probability"])
+    st.dataframe(df)
     
-# Display predictions as a pie chart
-st.write("### Prediction Chart")
-fig, ax = plt.subplots()
-ax.pie(df["Probability"], labels=df["Pokemon"], autopct='%1.1f%%', colors=plt.cm.Paired.colors)
-ax.set_title('Prediction Probabilities')
-st.pyplot(fig)
+    # Display predictions as a pie chart
+    st.write("### Prediction Chart")
+    fig, ax = plt.subplots()
+    ax.pie(df["Probability"], labels=df["Pokemon"], autopct='%1.1f%%', colors=plt.cm.Paired.colors)
+    ax.set_title('Prediction Probabilities')
+    st.pyplot(fig)