IBounhas commited on
Commit
d6b97a8
·
1 Parent(s): 5a59af5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ class CustomBertTokenizer(BertTokenizer):
11
  super(CustomBertTokenizer, self).__init__(*args, **kwargs)
12
  param_model_name="CAMeL-Lab/bert-base-arabic-camelbert-msa-sixteenth"
13
 
14
- tokenizer = CustomBertTokenizer.from_pretrained(param_model_name)
 
15
  class BertForSTS(torch.nn.Module):
16
 
17
  def __init__(self):
 
11
  super(CustomBertTokenizer, self).__init__(*args, **kwargs)
12
  param_model_name="CAMeL-Lab/bert-base-arabic-camelbert-msa-sixteenth"
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained(param_model_name)
15
+
16
  class BertForSTS(torch.nn.Module):
17
 
18
  def __init__(self):