wmzayed commited on
Commit
19a6f28
·
verified ·
1 Parent(s): 500aa3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,7 +27,8 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
27
  import torch
28
 
29
  # Load the tokenizer and model
30
- model_name = 'aubmindlab/bert-base-arabertv02'
 
31
  tokenizer = AutoTokenizer.from_pretrained(model_name)
32
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
33
 
 
27
  import torch
28
 
29
  # Load the tokenizer and model
30
+ # model_name = 'aubmindlab/bert-base-arabertv02'
31
+ model_name = 'PRAli22/AraBert-Arabic-Sentiment-Analysis'
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)
33
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
34