diff --git a/ensemble_inference.py b/ensemble_inference.py index 67f1119..18f252d 100644 --- a/ensemble_inference.py +++ b/ensemble_inference.py @@ -220,7 +220,7 @@ def main(): print('loading & stft of wave source...', end=' ') X_wave, y_wave, X_spec_s, y_spec_s = {}, {}, {}, {} - basename = os.path.splitext(os.path.basename(args.input))[0] + basename = '"{}"'.format(os.path.splitext(os.path.basename(args.input))[0]) bands_n = len(mp.param['band']) for d in range(bands_n, 0, -1): @@ -354,4 +354,4 @@ def main(): print('Complete!') if __name__ == '__main__': - main() \ No newline at end of file + main()