Update app.py

app.py

@@ -232,7 +232,7 @@ def apply_ocr(cell_coordinates, cropped_table):
     # return as Pandas dataframe
     df = pd.read_csv('output.csv')
 
-    return df
+    return df, data
 
 
 def process_pdf(image):
@@ -242,18 +242,18 @@ def process_pdf(image):
 
     cell_coordinates = get_cell_coordinates_by_row(cells)
 
-    data = apply_ocr(cell_coordinates, image)
+    df, data = apply_ocr(cell_coordinates, image)
 
-    return image, data
+    return image, df, data
     
 
 title = "Demo: table detection & recognition with Table Transformer"
 description = "Demo for table extraction with the Table Transformer (TATR)."
-examples =[['image.png', 'clip_paper.png']]
+examples = [['image.png', 'clip_paper.png']]
 
 app = gr.Interface(fn=process_pdf, 
                      inputs=gr.Image(type="pil"), 
-                     outputs=[gr.Image(type="pil", label="Detected table"), gr.Dataframe(label="Table as CSV")],
+                     outputs=[gr.Image(type="pil", label="Detected table"), gr.Dataframe(label="Table as CSV"), gr.JSON(label="Data as JSON")],
                      title=title,
                      description=description,
                      examples=examples)