Spaces:
Running
Running
Commit
·
1de09fd
1
Parent(s):
e66feb8
Update bertTagger
Browse files
app.py
CHANGED
|
@@ -24,8 +24,8 @@ with open("Nested/utils/tag_vocab.pkl", "rb") as f:
|
|
| 24 |
|
| 25 |
# model = torch.load(checkpoint_path, map_location="cpu")
|
| 26 |
model = BertSeqTagger(
|
| 27 |
-
|
| 28 |
-
|
| 29 |
)
|
| 30 |
|
| 31 |
|
|
|
|
| 24 |
|
| 25 |
# model = torch.load(checkpoint_path, map_location="cpu")
|
| 26 |
model = BertSeqTagger(
|
| 27 |
+
bert_model="aubmindlab/bert-base-arabertv2",
|
| 28 |
+
dropout=0.1
|
| 29 |
)
|
| 30 |
|
| 31 |
|