rishabh062 commited on
Commit
3d55008
1 Parent(s): 3579971

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ def process_document(image):
35
  # postprocess
36
  sequence = processor.batch_decode(outputs.sequences)[0]
37
  sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
38
- sequence = re.sub(r"<.*?>", "", sequence, count=1).strip() # remove first task start token
39
 
40
  return processor.token2json(sequence)
41
 
 
35
  # postprocess
36
  sequence = processor.batch_decode(outputs.sequences)[0]
37
  sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
38
+ sequence = re.sub(r"<.*?>", "", sequence, count=1).strip()
39
 
40
  return processor.token2json(sequence)
41