..

app.py

@@ -22,7 +22,7 @@ arabert_model = AutoModel.from_pretrained(arabert_model_name)
 # Load BERT for English topic classification
 bert_model_name = "bert-base-uncased"
 bert_tokenizer = AutoTokenizer.from_pretrained(bert_model_name)
-bert_model = BertForSequenceClassification.from_pretrained(bert_model_name, num_labels=3)  # Adjust labels as needed
+bert_model = BertForSequenceClassification.from_pretrained(bert_model_name, num_labels=2)  # Adjust labels as needed
 
 darija_topic_labels = ["Customer Service", "Retention Service", "Billing Issue"]  # Adjust for Darija topics
 english_topic_labels = ["Support Request", "Subscription Issue", "Payment Dispute"]  # Adjust for English topics