Update app.py

app.py

@@ -53,7 +53,12 @@ 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(f"Predicted Class: {class_labels[predicted_class]}")
+            st.text("Predicted Classes:")
+            for i, label in enumerate(class_labels):
+                if i == predicted_class:
+                    st.text(f"[**{label}**]")
+                else:
+                    st.text(label)
 
 elif option == "PDF":
     uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
@@ -74,4 +79,9 @@ 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(f"Predicted Class: {class_labels[predicted_class]}")
+                st.text("Predicted Classes:")
+                for i, label in enumerate(class_labels):
+                    if i == predicted_class:
+                        st.text(f"[**{label}**]")
+                    else:
+                        st.text(label)