Spaces:
Running on Zero
Running on Zero
| import json | |
| default_param = {} | |
| default_param["bins"] = -1 | |
| default_param["unstable_bins"] = -1 | |
| default_param["stable_bins"] = -1 | |
| default_param["sr"] = 44100 | |
| default_param["pre_filter_start"] = -1 | |
| default_param["pre_filter_stop"] = -1 | |
| default_param["band"] = {} | |
| N_BINS = "n_bins" | |
| def int_keys(d): | |
| result_dict = {} | |
| for key, value in d: | |
| if key.isdigit(): | |
| key = int(key) | |
| result_dict[key] = value | |
| return result_dict | |
| class ModelParameters(object): | |
| def __init__(self, model_params=None): | |
| self.param = model_params | |
| for k in [ | |
| "mid_side", | |
| "mid_side_b", | |
| "mid_side_b2", | |
| "stereo_w", | |
| "stereo_n", | |
| "reverse", | |
| ]: | |
| if not k in self.param: | |
| self.param[k] = False | |
| if N_BINS in self.param: | |
| self.param["bins"] = self.param[N_BINS] | |