Update app.py

app.py

@@ -291,16 +291,16 @@ def create_dataframe(cells_pytess_result:list, max_cols:int, max_rows:int,csv_pa
 
     return df
 
-def process_image(image, TD_THRESHOLD, TSR_THRESHOLD, padd_top, padd_left, padd_bottom, padd_right, delta_xmin, delta_ymin, delta_xmax, delta_ymax, expand_rowcol_bbox_top, expand_rowcol_bbox_bottom):
+def process_image(image, td_threshold, tsr_threshold, padd_top, padd_left, padd_bottom, padd_right, delta_xmin, delta_ymin, delta_xmax, delta_ymax, expand_rowcol_bbox_top, expand_rowcol_bbox_bottom):
   image = image.convert('RGB')
-  model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=TD_THRESHOLD)
+  model, probas, bboxes_scaled = table_detector(image, THRESHOLD_PROBA=td_threshold)
   # plot_results_detection(model, image, probas, bboxes_scaled,  delta_xmin, delta_ymin, delta_xmax, delta_ymax) 
   cropped_img_list = crop_tables(image, probas, bboxes_scaled, delta_xmin, delta_ymin, delta_xmax, delta_ymax)
 
   result = []
   for idx, unpadded_table in enumerate(cropped_img_list):
       table = add_padding(unpadded_table, padd_top, padd_right, padd_bottom, padd_left)
-      model, probas, bboxes_scaled = table_struct_recog(table, THRESHOLD_PROBA=TSR_THRESHOLD)
+      model, probas, bboxes_scaled = table_struct_recog(table, THRESHOLD_PROBA=tsr_threshold)
       rows, cols = generate_structure(model, table, probas, bboxes_scaled, expand_rowcol_bbox_top, expand_rowcol_bbox_bottom)
       rows, cols = sort_table_featuresv2(rows, cols)
       master_row, cols = individual_table_featuresv2(table, rows, cols)
@@ -323,10 +323,10 @@ article = "<p style='text-align: center'></p>"
 examples =[["image_0.png"]]
 
 iface = gr.Interface(fn=process_image, 
-                     inputs=[gr.Image(type="pil"), gr.Slider(0, 1, value=0.9), gr.Slider(0, 1, value=0.8), gr.Slider(0, 200, value=100), gr.Slider(0, 200, value=100), gr.Slider(0, 200, value=100), gr.Slider(0, 200, value=100), gr.Number(0), gr.Number(0), gr.Number(0), gr.Number(0),gr.Number(0),gr.Number(0)], 
+                     inputs=[gr.Image(type="pil"), gr.Slider(0, 1, 0.9), gr.Slider(0, 1, 0.8), gr.Slider(0, 200, 100), gr.Slider(0, 200, 100), gr.Slider(0, 200, 100), gr.Slider(0, 200, 100), gr.Number(0), gr.Number(0), gr.Number(0), gr.Number(0),gr.Number(0),gr.Number(0)], 
                      outputs="text",
                      title=title,
                      description=description,
                      article=article,
                      examples=examples)
-iface.launch(debug=True)
+iface.launch(debug=True)