ysharma HF staff commited on
Commit
2ab0f0e
1 Parent(s): a1b54b6

added examples

Browse files
Files changed (1) hide show
  1. app.py +15 -21
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
- #mkd = gr.Markdown(visible=False,scale=1)
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)