Update model.py

model.py

@@ -25,8 +25,8 @@ stopwords = stopwords_list()
 # Load the BERT model for sentiment analysis
 dataset = Dataset.from_pandas(pd.DataFrame({"Comment": []}))
 
-tokenizer = AutoTokenizer.from_pretrained("HooshvareLab/bert-fa-base-uncased")
-model = BertForSequenceClassification.from_pretrained("HooshvareLab/bert-fa-base-uncased", num_labels=3)
+tokenizer = AutoTokenizer.from_pretrained("HooshvareLab/bert-base-parsbert-uncased")
+model = BertForSequenceClassification.from_pretrained("HooshvareLab/bert-base-parsbert-uncased", num_labels=3)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)