Updated UI

Changed Predicted class and Confidence to display as df

app.py

@@ -93,10 +93,14 @@ def main():
                 with col2:
                     st.image(explanation_image, caption="Explanation Image", use_column_width=True)
                 
-                # Display prediction
+               # Display prediction
                 st.subheader("Prediction")
-                st.write(f"Predicted Class: {predicted_class}")
-                st.write(f"Confidence Score: {confidence_score}%")
+                # Create a table for prediction results
+                prediction_table = pd.DataFrame({
+                    "Predicted Class": [predicted_class],
+                    "Confidence": [f"{confidence_score}%"]
+                })
+                st.table(prediction_table)
 
 if __name__ == "__main__":
     main()