Update app.py

app.py

@@ -44,11 +44,7 @@ if uploaded_file:
     results_df = pd.DataFrame(
         {'Cell Type': categories, 'Probability': probs.tolist()}
     )
-
-    # Display the probabilities as a bar chart
-    st.bar_chart(results_df.set_index('Cell Type'))
-
-    # Highlight the most likely class
+     # Highlight the most likely class
     most_likely_class = categories[idx]
     st.success(f"Predicted Class: {most_likely_class}")
 
@@ -56,5 +52,9 @@ if uploaded_file:
     st.write("Detailed Classification Results:")
     st.table(results_df)
 
+    # Display the probabilities as a bar chart
+    st.bar_chart(results_df.set_index('Cell Type'))
+
+   
 else:
     st.warning("Upload an image to start classification.")