Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from streamlit_pills import pills
+#from streamlit_pills import pills
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import fitz
@@ -62,9 +62,9 @@ if option == "Text":
     affiliations_input = st.text_area("Enter Affiliations:")
     options=["Nursing", "Physics", "Maths", "Chemical", "Nuclear", "Engineering" ,"Other"]
     
-    categories = pills("Select WoS category", options)
+    #categories = pills("Select WoS category", options)
     
-    # categories = st.multiselect("Select WoS categories:", options)
+    categories = st.multiselect("Select WoS categories:", options)
 
     combined_text = f"{title_input} [SEP] {abstract_input} [SEP] {full_text_input} [SEP] {affiliations_input} [SEP] {' [SEP] '.join(categories)}"