Hugging Face's logo

Spaces:
saadiiii
/
ner-gradio
Runtime error

App Files Community
saadiiii commited on
Commit
8fda8d3
·
1 Parent(s): ac113ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ class BertModel(torch.nn.Module):
19
 
20
  super(BertModel, self).__init__()
21
 
22
- self.bert = BertForTokenClassification.from_pretrained('law-ai/InLegalBERT', num_labels=len(unique_labels))
23
 
24
  def forward(self, input_id, mask, label):
25
 
 
19
 
20
  super(BertModel, self).__init__()
21
 
22
+ self.bert = BertForTokenClassification.from_pretrained('law-ai/InLegalBERT', num_labels=14)
23
 
24
  def forward(self, input_id, mask, label):
25