Update app.py

app.py

@@ -44,6 +44,7 @@ class_colors = {
     3: "#d62728"   # Level 4
 }
 
+st.set_page_config("Paper Citation Calssifier")
 st.title("Paper Citation Classifier")
 
 option = st.radio("Select input type:", ("Text", "PDF"))
@@ -54,9 +55,10 @@ if option == "Text":
     abstract_input = st.text_area("Enter Abstract:")
     full_text_input = st.text_area("Enter Full Text:")
     affiliations_input = st.text_area("Enter Affiliations:")
-
-    # Select categories using pills
-    categories = st.multiselect("Select WoS categories:", ["Nursing", "Physics", "Maths", "Chemical", "Nuclear" ,"Other"])
+    options=["Nursing", "Physics", "Maths", "Chemical", "Nuclear" ,"Other"]
+    categories = pills("Select a category", options)
+    
+    # categories = st.multiselect("Select WoS categories:", ["Nursing", "Physics", "Maths", "Chemical", "Nuclear" ,"Other"])
 
     # Combine selected categories with [SEP]
     combined_text = f"{title_input} [SEP] {abstract_input} [SEP] {full_text_input} [SEP] {affiliations_input} [SEP] {' [SEP] '.join(categories)}"