Update app.py

app.py

@@ -9,8 +9,8 @@ longformer_model = AutoModelForSequenceClassification.from_pretrained("RREEM-ALR
 longformer_tokenizer = AutoTokenizer.from_pretrained("allenai/longformer-base-4096")
 
 # Load BERT model and tokenizer
-bert_model = AutoModelForSequenceClassification.from_pretrained("bert-base-uncased")
-bert_tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
+bert_model = AutoModelForSequenceClassification.from_pretrained("Mhassanen/bert-class-test")
+bert_tokenizer = AutoTokenizer.from_pretrained("Mhassanen/bert-class-test")
 
 # Function to extract text from PDF
 def extract_text_from_pdf(file_path):