Update app.py

app.py

@@ -175,16 +175,16 @@ model_synth = torch.load("synth2synth_full.pt", map_location="cpu").eval()
 
 
 
-def inference(aud, effect_type):
+def inference(aud, effect_type,gain_dB,c0,c1,mix,width,max_length):
   x_p, sample_rate = torchaudio.load(aud)
   
   effect_type = effect_type #@param ["Compressor", "Reverb", "Amp", "Analog Delay", "Synth2Synth"]
-  gain_dB = -24 #@param {type:"slider", min:-24, max:24, step:0.1}
-  c0 = -1.4 #@param {type:"slider", min:-10, max:10, step:0.1}
-  c1 = 3 #@param {type:"slider", min:-10, max:10, step:0.1}
-  mix = 70 #@param {type:"slider", min:0, max:100, step:1}
-  width = 50 #@param {type:"slider", min:0, max:100, step:1}
-  max_length = 30 #@param {type:"slider", min:5, max:120, step:1}
+  gain_dB = gain_dB #@param {type:"slider", min:-24, max:24, step:0.1}
+  c0 = c0 #@param {type:"slider", min:-10, max:10, step:0.1}
+  c1 = c1 #@param {type:"slider", min:-10, max:10, step:0.1}
+  mix = mix #@param {type:"slider", min:0, max:100, step:1}
+  width = width #@param {type:"slider", min:0, max:100, step:1}
+  max_length = max_length#@param {type:"slider", min:5, max:120, step:1}
   stereo = True #@param {type:"boolean"}
   tail = True #@param {type:"boolean"}
   
@@ -268,7 +268,11 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.029
 
 gr.Interface(
     inference, 
-    [gr.inputs.Audio(type="filepath", label="Input"),gr.inputs.Dropdown(choices=["Compressor", "Reverb", "Amp", "Analog Delay", "Synth2Synth"], type="value", default="Analog Delay", label="Effect Type")], 
+    [gr.inputs.Audio(type="filepath", label="Input"),gr.inputs.Dropdown(choices=["Compressor", "Reverb", "Amp", "Analog Delay", "Synth2Synth"], type="value", default="Analog Delay", label="Effect Type"),gr.inputs.Slider(minimum=-24, maximum=24, step=1, default=-24, label="gain dB"),gr.inputs.Slider(minimum=-10, maximum=10, step=0.1, default=-1.4, label="c0"),
+    gr.inputs.Slider(minimum=-10, maximum=10, step=1, default=3, label="c1"),
+    gr.inputs.Slider(minimum=0, maximum=100, step=1, default=70, label="mix"),
+    gr.inputs.Slider(minimum=0, maximum=100, step=1, default=50, label="width"),
+    gr.inputs.Slider(minimum=5, maximum=120, step=1, default=30, label="max length")], 
     gr.outputs.Audio(type="file", label="Output"),
     title=title,
     description=description,