v

app.py

@@ -20,12 +20,12 @@ import view
        
 def parse_args(args):
     parser = argparse.ArgumentParser(description="CBD Classification")
-    parser.add_argument('--data_dir', type=str, default="./")
-    parser.add_argument('--excel_file', type=str, default="dumc_1223_case3_duct_correct.csv")
+    parser.add_argument('--data_dir', type=str, default="./data/")
+    parser.add_argument('--excel_file', type=str, default="DUMC_final.csv")
     parser.add_argument('--mode', type=str, default="train")
     parser.add_argument('--scale', type=bool, default=True)
     parser.add_argument('--smote', type=bool, default=True)
-    parser.add_argument('--model_name_or_path', type=str, default="./FINAL", choices=[])
+    parser.add_argument('--model_name_or_path', type=str, default="./data/model", choices=[])
 
     return parser.parse_args(args)
 
@@ -142,7 +142,7 @@ if __name__ == '__main__':
         gr.Markdown(description)
         with gr.Row():
             with gr.Column():
-                tabular_input = gr.Dataframe(headers= tabular_header, datatype= tabular_dtype, label="Tabular Input", type="array", interactive=True, row_count=1, col_count=11)
+                tabular_input = gr.Dataframe(headers= tabular_header, datatype= tabular_dtype, label="Tabular Input", type="array", interactive=True, row_count=1, col_count=13)
                 info = gr.Textbox(lines=1, label="Patient info", visible = False)
 
                 with gr.Accordion("Parameters", open=False) as parameter_row: