Spaces:
Runtime error
Runtime error
Commit
·
76079cd
1
Parent(s):
ae44679
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
+
inputs = [gr.inputs.Audio(type="file", label="Input Audio"),gr.inputs.Radio(choices=['0','1','2'], type="value", default='0', label='mode')]
|
| 14 |
+
outputs = gr.outputs.Audio(type="file",label="Output Audio")
|
| 15 |
+
title = "Voice Fixer"
|
| 16 |
+
description = "Gradio demo for VoiceFixer: Toward General Speech Restoration With Neural Vocoder. To use it, simply add your audio, or click one of the examples to load them. Read more at the links below."
|
| 17 |
+
article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2109.13731' target='_blank'>VoiceFixer: Toward General Speech Restoration With Neural Vocoder</a> | <a href='https://github.com/haoheliu/voicefixer_main' target='_blank'>Github Repo</a></p>"
|
| 18 |
+
examples=[['bruce.wav','2']]
|
| 19 |
+
gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=examples, enable_queue=True).launch()
|