Update app.py

app.py

@@ -1,3 +1,46 @@
+Hugging Face's logo
+Hugging Face
+Search models, datasets, users...
+Models
+Datasets
+Spaces
+Docs
+Solutions
+Pricing
+
+
+
+Click here to log in through Single Sign-On to view activity within the huggingface org.
+
+Spaces:
+
+merve
+/
+nougat-transformers 
+
+like
+0
+
+App
+Files
+Community
+Settings
+nougat-transformers
+/
+app.py
+merve's picture
+merve
+HF STAFF
+Update app.py
+7c259fa
+7 minutes ago
+raw
+history
+blame
+edit
+delete
+No virus
+4.31 kB
 from huggingface_hub import hf_hub_download
 import re
 from PIL import Image
@@ -50,7 +93,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")
@@ -69,13 +112,12 @@ 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.txt","w+") as f:
+  with open(f"{os.getcwd()}/output.md","w+") as f:
       f.write(content)
       f.close()
-    
-  return content, f"{os.getcwd()}/output.txt"
+
+      
+  return content, f"{os.getcwd()}/output.md"
 
 
 css = """
@@ -101,8 +143,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(label='Download output as file 📑')
   with gr.Row():
     btn = gr.Button('Run Nougat 🍫')
   with gr.Row():
@@ -113,17 +153,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,
@@ -133,4 +173,4 @@ with gr.Blocks(css=css) as demo:
 
     
 demo.queue()
-demo.launch(debug=True)
+demo.launch(debug=True)