Spaces:

nickgambirasi
/

donut_text_extract

Running

nickgambirasi commited on Jul 13, 2023

Commit

2a0b298

1 Parent(s): 2703586

json return

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,8 +41,7 @@ def process_document(image):
     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
-    js = processor.token2json(sequence)
-    return pd.json_normalize(js)
 demo = gradio.Interface(
     fn=process_document,

     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
+    return processor.token2json(sequence)
 demo = gradio.Interface(
     fn=process_document,