..

app.py

@@ -24,7 +24,8 @@ translation_model = MarianMTModel.from_pretrained(translation_model_name).to(dev
 # AraBERT for Darija topic classification
 arabert_model_name = "aubmindlab/bert-base-arabert"
 arabert_tokenizer = AutoTokenizer.from_pretrained(arabert_model_name)
-arabert_model = AutoModel.from_pretrained(arabert_model_name).to(device)
+arabert_model = BertForSequenceClassification.from_pretrained(arabert_model_name, num_labels=2).to(device)
+
 
 # BERT for English topic classification
 bert_model_name = "bert-base-uncased"