update app

app.py

@@ -38,25 +38,18 @@ def process_document(image):
     # postprocess
     sequence = processor.batch_decode(outputs.sequences)[0]
     sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(processor.tokenizer.pad_token, "")
-    sequence = re.sub(r"<.*?>", "", sequence, count=1).strip()  # remove first task start token
+    sequence = re.sub(r"<.*?>", "", sequence).strip()  # remove first task start token
     
-    json = processor.token2json(sequence)
-    json_flatten = list(dict(**json).values())
-    out = ""
-    for value in json_flatten:
-        out += value + "\n"
-
-    return {'output': out}
+    return sequence
 
 demo = gradio.Interface(
     fn=process_document,
     inputs="image",
-    outputs="json",
+    outputs="string",
     title="Document Parsing Demo",
     description=None,
     article=None,
-    enable_queue=True,
     examples=None,
     cache_examples=False)
 
-demo.launch()
+demo.launch(enable_queue=True)