Hugging Face's logo

Spaces:
arosyihuddin
/
gradio-LegalNER
Sleeping

App Files Community
arosyihuddin commited on
Commit
1bf5976
·
1 Parent(s): 0ca7583

update UI

Browse files
Files changed (4) hide show
  1. model/IndoNLU/model_fold_4.pth +0 -3
  2. model/{IndoLEM/model_fold_4.pth → indoBERT-indoLEM-Fold-5.pth} +2 -2
  3. src/__pycache__/legalNER.cpython-311.pyc +0 -0
  4. src/legalNER.py +1 -1
model/IndoNLU/model_fold_4.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8e87a3141742561544df98e491338c11c7ff3d6ff7fcf67de5a76e44238f4222
3
- size 495563931
 
 
 
 
model/{IndoLEM/model_fold_4.pth → indoBERT-indoLEM-Fold-5.pth} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d378edcc860146fc622477232b07a3a3e5c618cb56b109c59c8af43b7cccd96
3
- size 440031387
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8e6bca094c31b725415a497779b1011c7a153d1f8399efa164e1a7c924c07b1
3
+ size 440031508
src/__pycache__/legalNER.cpython-311.pyc CHANGED
Binary files a/src/__pycache__/legalNER.cpython-311.pyc and b/src/__pycache__/legalNER.cpython-311.pyc differ
 
src/legalNER.py CHANGED
@@ -206,7 +206,7 @@ class LegalNER():
206
  if use_cuda:
207
  self.model = self.model.cuda()
208
 
209
- file_check_point = 'indoBERT-indoLEM-Fold-5.pth' if self.check_point == 'IndoBERT (IndoLEM)' else 'indoBERT-indoNLU-Fold-5.pth'
210
 
211
  model_weights = torch.load(file_check_point, map_location=torch.device(device))
212
  self.model.load_state_dict(model_weights)
 
206
  if use_cuda:
207
  self.model = self.model.cuda()
208
 
209
+ file_check_point = 'model/indoBERT-indoLEM-Fold-5.pth' if self.check_point == 'IndoBERT (IndoLEM)' else 'model/indoBERT-indoNLU-Fold-5.pth'
210
 
211
  model_weights = torch.load(file_check_point, map_location=torch.device(device))
212
  self.model.load_state_dict(model_weights)