nickgambirasi commited on
Commit
c7a67c9
·
1 Parent(s): 748364d

update output

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -40,7 +40,13 @@ def process_document(image):
40
  sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
41
  sequence = re.sub(r"<.*?>", "", sequence, count=1).strip() # remove first task start token
42
 
43
- return processor.token2json(sequence)
 
 
 
 
 
 
44
 
45
  demo = gradio.Interface(
46
  fn=process_document,
 
40
  sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
41
  sequence = re.sub(r"<.*?>", "", sequence, count=1).strip() # remove first task start token
42
 
43
+ json = processor.token2json(sequence)
44
+ json_flatten = dict(**json).values()
45
+ out = ""
46
+ for value in json_flatten:
47
+ out += value + "\n"
48
+
49
+ return out
50
 
51
  demo = gradio.Interface(
52
  fn=process_document,