Update app.py

app.py

@@ -256,15 +256,15 @@ def apply_ocr(cell_coordinates, cropped_table):
           row_data = row_data + ["" for _ in range(max_num_columns - len(row_data))]
         data[str(idx)] = row_data
 
-    # write to csv
-    with open('output.csv','w') as result_file:
-        wr = csv.writer(result_file, dialect='excel')
+    # # write to csv
+    # with open('output.csv','w') as result_file:
+    #     wr = csv.writer(result_file, dialect='excel')
     
-        for row, row_text in data.items():
-            wr.writerow(row_text)
+    #     for row, row_text in data.items():
+    #         wr.writerow(row_text)
 
-    # return as Pandas dataframe
-    df = pd.read_csv('output.csv')
+    # # return as Pandas dataframe
+    # df = pd.read_csv('output.csv')
 
     return data
 
@@ -276,9 +276,13 @@ def process_pdf(image):
 
     cell_coordinates = get_cell_coordinates_by_row(cells)
 
-    df, data = apply_ocr(cell_coordinates, image)
+    # df, data = apply_ocr(cell_coordinates, image)
 
-    return image, df, data
+    # return image, df, data
+    
+    data = apply_ocr(cell_coordinates, image)
+
+    return image, data
     
 
 title = "Demo: table detection & recognition with Table Transformer (TATR)."