Spaces:
Runtime error
Runtime error
added examples
Browse files
app.py
CHANGED
@@ -100,6 +100,10 @@ def predict1(pdf_file):
|
|
100 |
|
101 |
return content
|
102 |
|
|
|
|
|
|
|
|
|
103 |
css = """
|
104 |
#mkd {
|
105 |
height: 500px;
|
@@ -119,13 +123,12 @@ with gr.Blocks(css=css) as demo:
|
|
119 |
|
120 |
with gr.Row(equal_height=True):
|
121 |
pdf_file = gr.File(label='PDF📃', file_count='single', scale=1)
|
122 |
-
|
123 |
-
pdf_link = gr.Textbox(placeholder='Enter an arxiv link here', label='PDF link🔗🌐', scale=1)
|
124 |
|
125 |
with gr.Row():
|
126 |
btn = gr.Button('Run NOUGAT🍫')
|
127 |
clr = gr.Button('Clear🚿')
|
128 |
-
parsed_output = gr.Markdown(elem_id='mkd', value='OCR Output
|
129 |
|
130 |
btn.click(predict, [pdf_file, pdf_link], parsed_output )
|
131 |
clr.click(lambda : (gr.update(value=None),
|
@@ -135,23 +138,14 @@ with gr.Blocks(css=css) as demo:
|
|
135 |
[pdf_file, pdf_link, parsed_output]
|
136 |
)
|
137 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
138 |
demo.queue()
|
139 |
demo.launch(debug=True)
|
140 |
-
|
141 |
-
with gr.Blocks(css=css) as demo1:
|
142 |
-
gr.HTML("<h1><center>Nougat: Neural Optical Understanding for Academic Documents<center><h1>")
|
143 |
-
gr.HTML("<h3><center>Lukas Blecher et al. <a href='https://arxiv.org/pdf/2308.13418.pdf' target='_blank'>Paper</a>, <a href='https://facebookresearch.github.io/nougat/'>Project</a><center></h3>")
|
144 |
-
|
145 |
-
with gr.Row():
|
146 |
-
pdf_file = gr.File(label='Upload a PDF', scale=1)
|
147 |
-
mkd = gr.Markdown('<h2><center><i>OR</i></center></h2>',scale=1)
|
148 |
-
pdf_link = gr.Textbox(placeholder='Enter an arxiv link here', label='Provide a link', scale=1)
|
149 |
-
|
150 |
-
btn = gr.Button()
|
151 |
-
parsed_output = gr.Markdown(elem_id='mkd')
|
152 |
-
|
153 |
-
btn.click(predict, pdf_file, parsed_output )
|
154 |
-
|
155 |
-
#demo.queue()
|
156 |
-
#demo.launch(debug=True)
|
157 |
-
|
|
|
100 |
|
101 |
return content
|
102 |
|
103 |
+
def process_example(pdf_file,pdf_link):
|
104 |
+
ocr_content = paper_read(pdf_file,pdf_link)
|
105 |
+
return ocr_content
|
106 |
+
|
107 |
css = """
|
108 |
#mkd {
|
109 |
height: 500px;
|
|
|
123 |
|
124 |
with gr.Row(equal_height=True):
|
125 |
pdf_file = gr.File(label='PDF📃', file_count='single', scale=1)
|
126 |
+
pdf_link = gr.Textbox(placeholder='Enter an Arxiv link here', label='PDF link🔗🌐', scale=1)
|
|
|
127 |
|
128 |
with gr.Row():
|
129 |
btn = gr.Button('Run NOUGAT🍫')
|
130 |
clr = gr.Button('Clear🚿')
|
131 |
+
parsed_output = gr.Markdown(elem_id='mkd', value='📃🔤OCR Output')
|
132 |
|
133 |
btn.click(predict, [pdf_file, pdf_link], parsed_output )
|
134 |
clr.click(lambda : (gr.update(value=None),
|
|
|
138 |
[pdf_file, pdf_link, parsed_output]
|
139 |
)
|
140 |
|
141 |
+
gr.Examples(
|
142 |
+
[["input/nougat.pdf", ""], [None, "https://arxiv.org/pdf/2308.08316.pdf"]],
|
143 |
+
inputs = [pdf_file, pdf_link],
|
144 |
+
outputs = parsed_output,
|
145 |
+
fn=process_example,
|
146 |
+
cache_examples=True,
|
147 |
+
label='Click on any Examples below to get Nougat OCR results quickly:'
|
148 |
+
)
|
149 |
+
|
150 |
demo.queue()
|
151 |
demo.launch(debug=True)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|