update app.py

app.py

@@ -4,15 +4,16 @@ import os
 import utils
 import channel_mapping
 
-def test(model_name, fill_mode, csv_file, chanloc_file):
+def test(model_name, fill_mode, csv_file, chanlocs_file):
     tmp_path = os.path.dirname(str(csv_file))
     
     input_name = os.path.basename(str(csv_file))
     output_name = os.path.splitext(input_name)[0]+'_'+model_name+'.csv'
     
     # Channel mapping
-    #input_loc_name = os.path.basename(str(chanloc_file))
-    channel_mapping.mapping(csv_file, chanloc_file, fill_mode)
+    #input_loc_name = os.path.basename(str(chanlocs_file))
+    fill_mode = "adjacent" if fill_mode == "adjacent channel's value" else fill_mode
+    channel_mapping.mapping(csv_file, chanlocs_file, fill_mode)
     
     # step1: Data preprocessing
     total_file_num = utils.preprocessing(tmp_path, 'mapped.csv', 256)
@@ -20,18 +21,18 @@ def test(model_name, fill_mode, csv_file, chanloc_file):
     # step2: Signal reconstruction
     utils.reconstruct(model_name, total_file_num, tmp_path, output_name)
     
-    return tmp_path+'/'+output_data_name
+    return tmp_path+'/'+output_name
 
-input_data = gr.File(label="CSV file", file_types=[".csv"])
-input_loc = gr.File(label="Channel location", file_types=[".loc", "locs"])
+input_data = gr.File(label="Raw data (.csv)", file_types=[".csv"])
+input_loc = gr.File(label="Channel location (.loc, .locs)", file_types=[".loc", "locs"])
 input_model_name = gr.Dropdown(choices=["ICUNet", "UNetpp", "AttUnet", "EEGART"],
                                 value="ICUNet",
                                 label="Model")
-input_fill_mode = gr.Dropdown(choices=["zero", "adjacent"],
+input_fill_value = gr.Dropdown(choices=["zero", "adjacent channel's value"],
                                 value="zero",
-                                label="Fill empty channels with:")
+                                label="Imputation")
 
-inputs = [input_model_name, input_fill_mode, input_data, input_loc]
+inputs = [input_model_name, input_fill_value, input_data, input_loc]
 
 app = gr.Interface(fn=test,
                    inputs=inputs,
@@ -39,4 +40,4 @@ app = gr.Interface(fn=test,
                    allow_flagging="never")
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch(server_name="0.0.0.0", server_port=7860, share=False)