wmzayed commited on
Commit
458abd4
·
verified ·
1 Parent(s): 8b9ef2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +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
-
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)
33
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
34
 
 
27
  import torch
28
 
29
  # Load the tokenizer and model
30
+ # model_name = 'aubmindlab/bert-base-arabertv02'
31
+ model_name = 'aubmindlab/bert-base-arabertv02-twitter'
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)
33
  model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
34