Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 =
|
|
|
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):
|