Update app.py

app.py

@@ -29,14 +29,14 @@ def predict_pokemon(image):
     
     return probabilities_dict
 
-# Streamlit interface
-st.title("Pokemon Classifier")
-st.write("A simple MLP classification model for image classification using a pretrained model.")
+    # Streamlit interface
+    st.title("Pokemon Classifier")
+    st.write("A simple MLP classification model for image classification using a pretrained model.")
 
-# Upload image
-uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "png"])
+    # Upload image
+    uploaded_image = st.file_uploader("Choose an image...", type=["jpg", "png"])
 
-if uploaded_image is not None:
+    if uploaded_image is not None:
     image = Image.open(uploaded_image)
     st.image(image, caption='Uploaded Image.', use_column_width=True)
     st.write("")
@@ -50,8 +50,8 @@ if uploaded_image is not None:
     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)
+    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)