Ogpoggi commited on
Commit
b4cc034
·
1 Parent(s): d2651a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import re
2
- import gradio as gr
3
 
4
  import torch
5
  from transformers import DonutProcessor, VisionEncoderDecoderModel
@@ -39,18 +39,18 @@ def process_document(image):
39
 
40
  return processor.token2json(sequence)
41
 
42
- description = "Gradio Demo for Donut, an instance of `VisionEncoderDecoderModel` fine-tuned on CORD (document parsing). To use it, simply upload your image and click 'submit', or click one of the examples to load them. Read more at the links below."
43
- article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.15664' target='_blank'>Donut: OCR-free Document Understanding Transformer</a> | <a href='https://github.com/clovaai/donut' target='_blank'>Github Repo</a></p>"
44
-
45
- demo = gr.Interface(
46
- fn=process_document,
47
- inputs="image",
48
- outputs="json",
49
- title="Demo: Donut 🍩 for Document Parsing",
50
- description=description,
51
- article=article,
52
- enable_queue=True,
53
- examples=[["example.png"], ["example_1.png"],["example_2.png"], ["example_3.png"],["example_4.png"]],
54
- cache_examples=False)
55
-
56
- demo.launch()
 
1
  import re
2
+
3
 
4
  import torch
5
  from transformers import DonutProcessor, VisionEncoderDecoderModel
 
39
 
40
  return processor.token2json(sequence)
41
 
42
+ # description = "Gradio Demo for Donut, an instance of `VisionEncoderDecoderModel` fine-tuned on CORD (document parsing). To use it, simply upload your image and click 'submit', or click one of the examples to load them. Read more at the links below."
43
+ # article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.15664' target='_blank'>Donut: OCR-free Document Understanding Transformer</a> | <a href='https://github.com/clovaai/donut' target='_blank'>Github Repo</a></p>"
44
+
45
+ # demo = gr.Interface(
46
+ # fn=process_document,
47
+ # inputs="image",
48
+ # outputs="json",
49
+ # title="Demo: Donut 🍩 for Document Parsing",
50
+ # description=description,
51
+ # article=article,
52
+ # enable_queue=True,
53
+ # examples=[["example.png"], ["example_1.png"],["example_2.png"], ["example_3.png"],["example_4.png"]],
54
+ # cache_examples=False)
55
+
56
+ # demo.launch()