Update app.py

app.py

@@ -53,12 +53,7 @@ if option == "Text":
         predicted_class = predict_class(combined_text)
         if predicted_class is not None:
             class_labels = ["Level 1", "Level 2", "Level 3", "Level 4"]
-            st.text("Predicted Classes:")
-            for i, label in enumerate(class_labels):
-                if i == predicted_class:
-                    st.markdown(f'<div style="background-color: #ffa07a; padding: 5px;"><strong>{label}</strong></div>', unsafe_allow_html=True)
-                else:
-                    st.markdown(f'<div>{label}</div>', unsafe_allow_html=True)
+            st.text(f"Predicted Class: {class_labels[predicted_class]}")
 
 elif option == "PDF":
     uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
@@ -79,9 +74,4 @@ elif option == "PDF":
             predicted_class = predict_class(file_text)
             if predicted_class is not None:
                 class_labels = ["Level 1", "Level 2", "Level 3", "Level 4"]
-                st.text("Predicted Classes:")
-                for i, label in enumerate(class_labels):
-                    if i == predicted_class:
-                        st.markdown(f'<div style="background-color: #ffa07a; padding: 5px;"><strong>{label}</strong></div>', unsafe_allow_html=True)
-                    else:
-                        st.markdown(f'<div>{label}</div>', unsafe_allow_html=True)
+                st.text(f"Predicted Class: {class_labels[predicted_class]}")