update

app.py

@@ -832,7 +832,7 @@ with gr.Blocks(js=js, delete_cache=(3600, 3600)) as demo:
 				batch_md : gr.Markdown("", visible=True),
 				out_data_file : gr.File(value=None, visible=False)}
 	
-	def run_model(stage1_info, stage2_info, samplerate, modelname):
+	def run_model(stage1_info, stage2_info, channel_info, samplerate, modelname):
 		if stage2_info["errorFlag"] == True:
 			stage2_info["errorFlag"] = False
 			yield {stage2_json : stage2_info}
@@ -841,14 +841,15 @@ with gr.Blocks(js=js, delete_cache=(3600, 3600)) as demo:
 			filepath = stage2_info["filePath"]
 			inputname = stage2_info["fileNames"]["inputData"]
 			outputname = stage2_info["fileNames"]["outputData"]
+			channel_num = len(channel_info["inputNames"])
 			mapping_result = stage1_info["mappingResult"]
-			break_flag = False
 			
+			break_flag = False
 			for i in range(stage1_info["batch"]):
 				yield {batch_md : gr.Markdown('Running model({}/{})...'.format(i+1, stage1_info["batch"]))}
 				try:
 					# step1: Data preprocessing
-					preprocess_data, channel_num = utils.preprocessing(filepath, inputname, int(samplerate), mapping_result[i])
+					preprocess_data = utils.preprocessing(filepath, inputname, int(samplerate), mapping_result[i])
 					# step2: Signal reconstruction
 					reconstructed_data = utils.reconstruct(modelname, preprocess_data, filepath, i)
 					# step3: Data postprocessing
@@ -874,7 +875,7 @@ with gr.Blocks(js=js, delete_cache=(3600, 3600)) as demo:
 		outputs = [stage2_json, run_btn, cancel_btn, batch_md, out_data_file]
 	).success(
 		fn = run_model,
-		inputs = [stage1_json, stage2_json, in_samplerate, in_modelname],
+		inputs = [stage1_json, stage2_json, channel_json, in_samplerate, in_modelname],
 		outputs = [stage2_json, run_btn, cancel_btn, batch_md, out_data_file]
 	)
 	

app_utils.py

@@ -313,6 +313,7 @@ def mapping_result(stage1_info, channel_info, filename):
 	data = {
 		#"templateNames" : channel_info["templateNames"],
 		#"inputNames" : channel_info["inputNames"],
+		"channelNum" : len(channel_info["inputNames"]),
 		"batch" : batch,
 		"mappingResult" : results
 	}

utils.py

@@ -202,7 +202,7 @@ def preprocessing(filepath, inputfile, samplerate, mapping_result):
     
     # read data
     signal = read_train_data(inputfile)
-    channel_num = signal.shape[0]
+    #print(signal.shape)
     # channel mapping
     signal = reorder_data(signal, mapping_result)
     #print(signal.shape)
@@ -215,7 +215,7 @@ def preprocessing(filepath, inputfile, samplerate, mapping_result):
     # cutting data
     total_file_num = cut_data(filepath, signal)
 
-    return total_file_num, channel_num
+    return total_file_num
 
 def restore_order(data, all_data, mapping_result):
     for i, (indices, flag) in enumerate(zip(mapping_result["index"], mapping_result["isOriginalData"])):