Spaces:
Running
Running
modify fx norm
Browse files- inference.py +7 -7
inference.py
CHANGED
|
@@ -165,7 +165,7 @@ class MasteringStyleTransfer:
|
|
| 165 |
return "No parameters available"
|
| 166 |
|
| 167 |
param_mapper = {
|
| 168 |
-
'
|
| 169 |
'low_shelf_gain_db': ('Low Shelf Gain', 'dB', -20, 20),
|
| 170 |
'low_shelf_cutoff_freq': ('Low Shelf Cutoff', 'Hz', 20, 2000),
|
| 171 |
'low_shelf_q_factor': ('Low Shelf Q', '', 0.1, 5.0),
|
|
@@ -185,11 +185,11 @@ class MasteringStyleTransfer:
|
|
| 185 |
'high_shelf_cutoff_freq': ('High Shelf Cutoff', 'Hz', 4000, 20000), # Assuming sample_rate is 44100
|
| 186 |
'high_shelf_q_factor': ('High Shelf Q', '', 0.1, 5.0),
|
| 187 |
},
|
| 188 |
-
'
|
| 189 |
'drive_db': ('Drive', 'dB', 0, 8),
|
| 190 |
'parallel_weight_factor': ('Dry/Wet Mix', '%', 0, 100),
|
| 191 |
},
|
| 192 |
-
'
|
| 193 |
'low_cutoff': ('Low/Mid Crossover', 'Hz', 20, 1000),
|
| 194 |
'high_cutoff': ('Mid/High Crossover', 'Hz', 1000, 20000),
|
| 195 |
'parallel_weight_factor': ('Dry/Wet Mix', '%', 0, 100),
|
|
@@ -212,13 +212,13 @@ class MasteringStyleTransfer:
|
|
| 212 |
'high_shelf_at': ('High Band Attack Time', 'ms', 5, 100),
|
| 213 |
'high_shelf_rt': ('High Band Release Time', 'ms', 5, 100),
|
| 214 |
},
|
| 215 |
-
'
|
| 216 |
'gain_db': ('Output Gain', 'dB', -24, 24),
|
| 217 |
},
|
| 218 |
-
'
|
| 219 |
'width': ('Stereo Width', '', 0, 1),
|
| 220 |
},
|
| 221 |
-
'
|
| 222 |
'threshold': ('Threshold', 'dB', -60, 0),
|
| 223 |
'at': ('Attack Time', 'ms', 5, 100),
|
| 224 |
'rt': ('Release Time', 'ms', 5, 100),
|
|
@@ -227,7 +227,7 @@ class MasteringStyleTransfer:
|
|
| 227 |
|
| 228 |
output = []
|
| 229 |
for fx_name, fx_params in params.items():
|
| 230 |
-
output.append(f"{fx_name}:")
|
| 231 |
if isinstance(fx_params, dict):
|
| 232 |
for param_name, param_value in fx_params.items():
|
| 233 |
if isinstance(param_value, torch.Tensor):
|
|
|
|
| 165 |
return "No parameters available"
|
| 166 |
|
| 167 |
param_mapper = {
|
| 168 |
+
'eq': {
|
| 169 |
'low_shelf_gain_db': ('Low Shelf Gain', 'dB', -20, 20),
|
| 170 |
'low_shelf_cutoff_freq': ('Low Shelf Cutoff', 'Hz', 20, 2000),
|
| 171 |
'low_shelf_q_factor': ('Low Shelf Q', '', 0.1, 5.0),
|
|
|
|
| 185 |
'high_shelf_cutoff_freq': ('High Shelf Cutoff', 'Hz', 4000, 20000), # Assuming sample_rate is 44100
|
| 186 |
'high_shelf_q_factor': ('High Shelf Q', '', 0.1, 5.0),
|
| 187 |
},
|
| 188 |
+
'distortion': {
|
| 189 |
'drive_db': ('Drive', 'dB', 0, 8),
|
| 190 |
'parallel_weight_factor': ('Dry/Wet Mix', '%', 0, 100),
|
| 191 |
},
|
| 192 |
+
'multiband_comp': {
|
| 193 |
'low_cutoff': ('Low/Mid Crossover', 'Hz', 20, 1000),
|
| 194 |
'high_cutoff': ('Mid/High Crossover', 'Hz', 1000, 20000),
|
| 195 |
'parallel_weight_factor': ('Dry/Wet Mix', '%', 0, 100),
|
|
|
|
| 212 |
'high_shelf_at': ('High Band Attack Time', 'ms', 5, 100),
|
| 213 |
'high_shelf_rt': ('High Band Release Time', 'ms', 5, 100),
|
| 214 |
},
|
| 215 |
+
'gain': {
|
| 216 |
'gain_db': ('Output Gain', 'dB', -24, 24),
|
| 217 |
},
|
| 218 |
+
'imager': {
|
| 219 |
'width': ('Stereo Width', '', 0, 1),
|
| 220 |
},
|
| 221 |
+
'limiter': {
|
| 222 |
'threshold': ('Threshold', 'dB', -60, 0),
|
| 223 |
'at': ('Attack Time', 'ms', 5, 100),
|
| 224 |
'rt': ('Release Time', 'ms', 5, 100),
|
|
|
|
| 227 |
|
| 228 |
output = []
|
| 229 |
for fx_name, fx_params in params.items():
|
| 230 |
+
output.append(f"{fx_name.upper()}:")
|
| 231 |
if isinstance(fx_params, dict):
|
| 232 |
for param_name, param_value in fx_params.items():
|
| 233 |
if isinstance(param_value, torch.Tensor):
|