Update app.py

app.py

@@ -8,10 +8,10 @@ def inference(audio,mode):
     voicefixer.restore(input=audio.name, # input wav file path
                     output="output.wav", # output wav file path
                     cuda=False, # whether to use gpu acceleration
-                    mode = mode) # You can try out mode 0, 1 to find out the best result
+                    mode = int(mode)) # You can try out mode 0, 1 to find out the best result
     return 'output.wav'
 
-inputs = [gr.inputs.Audio(type="file", label="Input Audio"),gr.inputs.Slider(minimum=0, maximum=1, step=0.1, default=0, label='mode')]
+inputs = [gr.inputs.Audio(type="file", label="Input Audio"),gr.inputs.Radio(choices=['0','1','2'], type="value", default='0', label='mode')]
 outputs =  gr.outputs.Audio(type="file",label="Output Audio")