Update app.py
Browse files
app.py
CHANGED
|
@@ -32,13 +32,13 @@ def nougat_ocr(file_name):
|
|
| 32 |
subprocess.run(cli_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
| 33 |
|
| 34 |
def predict(pdf_file, pdf_link):
|
| 35 |
-
if pdf_file is None and
|
| 36 |
return "No data provided. Upload a pdf file or provide a pdf link and try again!", "", ""
|
| 37 |
|
| 38 |
-
if pdf_file is None:
|
| 39 |
-
file_name = get_pdf(pdf_link)
|
| 40 |
-
else:
|
| 41 |
file_name = pdf_file.name
|
|
|
|
|
|
|
| 42 |
|
| 43 |
# Call nougat
|
| 44 |
nougat_ocr(file_name)
|
|
@@ -77,9 +77,9 @@ with gr.Blocks(css=css) as demo:
|
|
| 77 |
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>")
|
| 78 |
|
| 79 |
with gr.Row():
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
|
| 84 |
with gr.Row(equal_height=True):
|
| 85 |
pdf_file = gr.File(label='PDF📃', file_count='single', scale=1)
|
|
@@ -96,7 +96,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 96 |
download_link = gr.File(elem_id='download-link', visible=False)
|
| 97 |
|
| 98 |
btn.click(fn=predict, inputs=[pdf_file, pdf_link], outputs=[parsed_output, markdown_hidden, download_link], _js=js_auto_copy)
|
| 99 |
-
clr.click(lambda: (gr.update(value=None), gr.update(value=None), gr.update(value=None)), [], [pdf_file, pdf_link, parsed_output, markdown_hidden, download_link])
|
| 100 |
dwn.click(None, [download_link], None)
|
| 101 |
|
| 102 |
# Enable queueing for request handling
|
|
|
|
| 32 |
subprocess.run(cli_command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
| 33 |
|
| 34 |
def predict(pdf_file, pdf_link):
|
| 35 |
+
if pdf_file is None and not pdf_link:
|
| 36 |
return "No data provided. Upload a pdf file or provide a pdf link and try again!", "", ""
|
| 37 |
|
| 38 |
+
if pdf_file is not None:
|
|
|
|
|
|
|
| 39 |
file_name = pdf_file.name
|
| 40 |
+
else:
|
| 41 |
+
file_name = get_pdf(pdf_link)
|
| 42 |
|
| 43 |
# Call nougat
|
| 44 |
nougat_ocr(file_name)
|
|
|
|
| 77 |
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>")
|
| 78 |
|
| 79 |
with gr.Row():
|
| 80 |
+
gr.Markdown('<h4><center>Upload a PDF</center></h4>', scale=1)
|
| 81 |
+
gr.Markdown('<h4><center><i>OR</i></center></h4>', scale=1)
|
| 82 |
+
gr.Markdown('<h4><center>Provide a PDF link</center></h4>', scale=1)
|
| 83 |
|
| 84 |
with gr.Row(equal_height=True):
|
| 85 |
pdf_file = gr.File(label='PDF📃', file_count='single', scale=1)
|
|
|
|
| 96 |
download_link = gr.File(elem_id='download-link', visible=False)
|
| 97 |
|
| 98 |
btn.click(fn=predict, inputs=[pdf_file, pdf_link], outputs=[parsed_output, markdown_hidden, download_link], _js=js_auto_copy)
|
| 99 |
+
clr.click(lambda: (gr.update(value=None), gr.update(value=None), gr.update(value=None), gr.update(value=None)), [], [pdf_file, pdf_link, parsed_output, markdown_hidden, download_link])
|
| 100 |
dwn.click(None, [download_link], None)
|
| 101 |
|
| 102 |
# Enable queueing for request handling
|