Update app.py

app.py

@@ -3,12 +3,13 @@ import subprocess
 
 
 def nougat_ocr(file_name):
+  print('******* inside nougat_ocr *******')
   # CLI Command to run
   cli_command = [
       'nougat',
-      '--out', '/output',
+      '--out', 'output',
       'pdf', f'{file_name}',
-      '--checkpoint', '/nougat'
+      '--checkpoint', 'nougat'
   ]
 
   # Run the command and get .mmd file in an output folder
@@ -17,23 +18,25 @@ def nougat_ocr(file_name):
 
 
 def predict(pdf_file):
+  print('******* inside predict *******')
   print(f"temporary file - {pdf_file.name}")
   pdf_name = pdf_file.name.split('/')[-1].split('.')[0]
   print(f"pdf file name - {pdf_name}")
     
   #! Get prediction for a PDF using nougat
   nougat_ocr(pdf_file.name)
+  print("BAACCKKK")
   
   # Open the multimarkdown (.mmd) file for reading
-  with open(f'/output/{pdf_name}.mmd', 'r') as file:
+  with open(f'output/{pdf_name}.mmd', 'r') as file:
       content = file.read() 
       
   return content
 
 
 with gr.Blocks() as demo:
-  gr.HTML("<h1>Nougat: Neural Optical Understanding for Academic Documents<h1>")
-  gr.HTML("<h3>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></h3>")
+  gr.HTML("<h1><center>Nougat: Neural Optical Understanding for Academic Documents<center><h1>")
+  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>")
   
   with gr.Row():
     pdf_file = gr.File(label='Upload a PDF', scale=1)