Reem333 commited on
Commit
dbdba7b
1 Parent(s): fe9d5fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
  import fitz
5
  import os
6
 
7
- model = AutoModelForSequenceClassification.from_pretrained("REEM-ALRASHIDI/LongFormer-Paper-Citaion-Classifier")
8
  tokenizer = AutoTokenizer.from_pretrained("allenai/longformer-base-4096")
9
 
10
  def extract_text_from_pdf(file_path):
 
4
  import fitz
5
  import os
6
 
7
+ model = AutoModelForSequenceClassification.from_pretrained("Reem333/LongFormer-Paper-Citaion-Classifier")
8
  tokenizer = AutoTokenizer.from_pretrained("allenai/longformer-base-4096")
9
 
10
  def extract_text_from_pdf(file_path):