Kevin676 commited on
Commit
3ab9aac
1 Parent(s): 7547bc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,17 +1,18 @@
1
  import os
 
2
  os.system('pip install voicefixer --upgrade')
3
  from voicefixer import VoiceFixer
4
  import gradio as gr
5
  voicefixer = VoiceFixer()
6
  def inference(audio,mode):
7
- voicefixer.restore(input=audio, # input wav file path
8
  output="output.wav", # output wav file path
9
  cuda=False, # whether to use gpu acceleration
10
  mode = int(mode)) # You can try out mode 0, 1 to find out the best result
11
  return 'output.wav'
12
 
13
- inputs = [gr.inputs.Audio(type="filepath", label="Input Audio"),gr.inputs.Radio(choices=['0','1','2'], type="value", default='0', label='mode')]
14
- outputs = gr.outputs.Audio(type="filepath",label="Output Audio")
15
 
16
 
17
  title = "Voice Fixer"
@@ -20,4 +21,4 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2109.137
20
 
21
  examples=[['bruce.wav','2']]
22
 
23
- gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=examples, enable_queue=True).launch()
 
1
  import os
2
+ os.system('pip install gradio==2.3.0a0')
3
  os.system('pip install voicefixer --upgrade')
4
  from voicefixer import VoiceFixer
5
  import gradio as gr
6
  voicefixer = VoiceFixer()
7
  def inference(audio,mode):
8
+ voicefixer.restore(input=audio.name, # input wav file path
9
  output="output.wav", # output wav file path
10
  cuda=False, # whether to use gpu acceleration
11
  mode = int(mode)) # You can try out mode 0, 1 to find out the best result
12
  return 'output.wav'
13
 
14
+ inputs = [gr.inputs.Audio(type="file", label="Input Audio"),gr.inputs.Radio(choices=['0','1','2'], type="value", default='0', label='mode')]
15
+ outputs = gr.outputs.Audio(type="file",label="Output Audio")
16
 
17
 
18
  title = "Voice Fixer"
 
21
 
22
  examples=[['bruce.wav','2']]
23
 
24
+ gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=examples, enable_queue=True).launch()