Update app.py

app.py

@@ -6,6 +6,13 @@ import os
 
 model = AutoModelForSequenceClassification.from_pretrained("Reem333/Citaion-Classifier")
 tokenizer = AutoTokenizer.from_pretrained("allenai/longformer-base-4096")
+def extract_text_from_pdf(file_path):
+    text = ''
+    with fitz.open(file_path) as pdf_document:
+        for page_number in range(pdf_document.page_count):
+            page = pdf_document.load_page(page_number)
+            text += page.get_text()
+    return text
 
 def predict_class(text):
     try:
@@ -43,7 +50,7 @@ with st.sidebar:
     st.markdown("## About")
     st.markdown('''
     This is a tool to classify paper citations into different levels based on their number of citations.
-    Powered by Fine-Tuned [Longformer model](https://huggingface.co/REEM-ALRASHIDI/LongFormer-Paper-Citaion-Classifier) with custom data.
+    Powered by Fine-Tuned [Longformer model](https://huggingface.co/Reem333/LongFormer-Paper-Citaion-Classifier) with custom data.
     ''')
     st.markdown("### Class Levels:")
     st.markdown("- Level 1: Highly cited papers")