Spaces:

rishabh062
/

donutCordImgToCsv

Runtime error

rishabh062 commited on Apr 26, 2023

Commit

3d55008

1 Parent(s): 3579971

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,7 @@ 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
     return processor.token2json(sequence)

     # 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()
     return processor.token2json(sequence)