Update app.py

app.py

@@ -175,7 +175,7 @@ model_synth = torch.load("synth2synth_full.pt", map_location="cpu").eval()
 
 
 
-def inference(aud, effect_type,gain_dB,c0,c1,mix,width,max_length):
+def inference(aud, effect_type,gain_dB,c0,c1,mix,width,max_length,stereo,tail):
   x_p, sample_rate = torchaudio.load(aud)
   
   effect_type = effect_type #@param ["Compressor", "Reverb", "Amp", "Analog Delay", "Synth2Synth"]
@@ -185,8 +185,8 @@ def inference(aud, effect_type,gain_dB,c0,c1,mix,width,max_length):
   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"}
+  stereo = stereo #@param {type:"boolean"}
+  tail = tail #@param {type:"boolean"}
   
   # select model type
   if effect_type == "Compressor":
@@ -272,7 +272,9 @@ gr.Interface(
     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.inputs.Slider(minimum=5, maximum=120, step=1, default=30, label="max length"),
+    gr.inputs.Checkbox(default=True, label="stereo"),
+    gr.inputs.Checkbox(default=True, label="tail")], 
     gr.outputs.Audio(type="file", label="Output"),
     title=title,
     description=description,