TymaaHammouda commited on
Commit
1de09fd
·
1 Parent(s): e66feb8

Update bertTagger

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- pretrained_path="aubmindlab/bert-base-arabertv2",
28
- dropout_p=0.1
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