Hugging Face's logo

Spaces:
NiamaLynn
/
lilt-roberta-DocLayNet-256
Runtime error

App Files Community
NiamaLynn commited on
Commit
18dfb41
·
1 Parent(s): 475780e

Update files/functions.py

Browse files
Files changed (1) hide show
  1. files/functions.py +2 -2
files/functions.py CHANGED
@@ -69,8 +69,8 @@ label2id = model.config.label2id
69
  num_labels = len(id2label)
70
 
71
  # (tokenization) The maximum length of a feature (sequence)
72
- if str(384) in model_id:
73
- max_length = 384
74
  elif str(512) in model_id:
75
  max_length = 512
76
  else:
 
69
  num_labels = len(id2label)
70
 
71
  # (tokenization) The maximum length of a feature (sequence)
72
+ if str(256) in model_id:
73
+ max_length = 256
74
  elif str(512) in model_id:
75
  max_length = 512
76
  else: