F-allahmoradi commited on
Commit
297b054
·
verified ·
1 Parent(s): 02f9da2

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -28,7 +28,7 @@ dataset = Dataset.from_pandas(pd.DataFrame({"Comment": []}))
28
 
29
 
30
  # بارگذاری مدل و توکنایزر
31
- model_name = "AutoModelForSequenceClassification.from_pretrained"
32
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=5, ignore_mismatched_sizes=True)
33
  tokenizer = AutoTokenizer.from_pretrained(model_name)
34
 
 
28
 
29
 
30
  # بارگذاری مدل و توکنایزر
31
+ model_name = "HooshvareLab/bert-fa-base-uncased-sentiment-deepsentipers-multi"
32
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=5, ignore_mismatched_sizes=True)
33
  tokenizer = AutoTokenizer.from_pretrained(model_name)
34