Update app.py

app.py

@@ -50,7 +50,7 @@ def predict(image):
 
 
 
-def inference(pdf_file, pdf_link, file_btn):
+def inference(pdf_file, pdf_link):
   if pdf_file is None:
     if pdf_link == '':
       print("No file is uploaded and No link is provided")
@@ -70,15 +70,13 @@ def inference(pdf_file, pdf_link, file_btn):
 
   content = sequence.replace(r'\(', '$').replace(r'\)', '$').replace(r'\[', '$$').replace(r'\]', '$$')
 
-  if file_btn:
+
     with open(f"{os.getcwd()}/output.md","w+") as f:
       f.write(content)
       f.close()
-    file_path = f"{os.getcwd()}/output.md"
-  else:
-      file_path=""
+
       
-  return content, file_path
+  return content, f"{os.getcwd()}/output.md"
 
 
 css = """
@@ -104,8 +102,6 @@ with gr.Blocks(css=css) as demo:
   with gr.Row(equal_height=True):
     pdf_file = gr.File(label='PDF 📑', file_count='single', scale=1)
     pdf_link = gr.Textbox(placeholder='Enter an arxiv link here', label='Link to Paper🔗', scale=1)
-  with gr.Row():
-    file_btn = gr.Checkbox(value=True, label='Download output as file 📑')
   with gr.Row():
     btn = gr.Button('Run Nougat 🍫')
   with gr.Row():
@@ -116,17 +112,17 @@ with gr.Blocks(css=css) as demo:
       parsed_output = gr.Markdown(elem_id='mkd', value='Output Text 📝')
       output_file = gr.File(file_types = ["txt"], label="Output File 📑")
   
-  btn.click(inference, [pdf_file, pdf_link, file_btn], [parsed_output, output_file])
+  btn.click(inference, [pdf_file, pdf_link], [parsed_output, output_file])
   clr.click(lambda : (gr.update(value=None), 
                       gr.update(value=None),
                       gr.update(value=None), 
                       gr.update(value=None)), 
              [], 
-             [pdf_file, pdf_link, file_btn, parsed_output, output_file]
+             [pdf_file, pdf_link, parsed_output, output_file]
             )
   gr.Examples(
-      [["nougat.pdf", "", True], [None, "https://arxiv.org/pdf/2308.08316.pdf", True]],
-      inputs = [pdf_file, pdf_link, file_btn],
+      [["nougat.pdf", ""], [None, "https://arxiv.org/pdf/2308.08316.pdf"]],
+      inputs = [pdf_file, pdf_link],
       outputs = [parsed_output, output_file],
       fn=inference,
       cache_examples=True,