Update app.py

app.py

@@ -48,7 +48,7 @@ st.set_page_config(page_title="Paper Citation Classifier", page_icon="logo.png")
 
 with st.sidebar:
     st.image("logo.png", width=70)
-    st.markdown('<div style="position: absolute; top: 5px; left: 5px;"></div>', unsafe_allow_html=True)
+    st.markdown('<div style="position: absolute; left: 5px;"></div>', unsafe_allow_html=True)
     
     st.markdown("# Paper Citation Classifier")
     st.markdown("---")
@@ -89,7 +89,7 @@ if option == "Text":
             with st.spinner("Predicting..."):
                 predicted_class = predict_class(combined_text)
                 if predicted_class is not None:
-                    class_labels = ["Level 1 (Highly Cited Paper)", "Level 2 (Average Cited Paper)", "Level 3 (More Cited Paper)", "Level 4 (Low Cited Paper)"]
+                    class_labels = ["Level 1", "Level 2", "Level 3", "Level 4"]
 
                     st.text("Predicted Class:")
                     for i, label in enumerate(class_labels):
@@ -123,7 +123,7 @@ elif option == "PDF":
                     with st.spinner("Predicting..."):
                         predicted_class = predict_class(file_text)
                         if predicted_class is not None:
-                            class_labels = ["Level 1 (Highly Cited Paper)", "Level 2 (Average Cited Paper)", "Level 3 (More Cited Paper)", "Level 4 (Low Cited Paper)"]
+                            class_labels = ["Level 1", "Level 2", "Level 3", "Level 4"]
                             st.text("Predicted Class:")
                             for i, label in enumerate(class_labels):
                                 if i == predicted_class: