import gradio as gr
from audio_style_transfer.models import timedomain

def audioStyleTransfer(content,style):
    print (style,content)
    output = "/tmp/outputfile.wav"
    timedomain.run( style_fname=style, content_fname=content ,output_fname=output, n_fft=4096,
               n_filters=4096,
               hop_length=512,
               alpha=0.05,
               k_w=4)
    print ("output is " ,output)
    return  output

iface = gr.Interface(fn=audioStyleTransfer, inputs=[gr.Audio(source="upload",type="filepath",label="Content"),gr.Audio(source="upload",type="filepath",label="Style")], outputs=gr.Audio(label="Output"))
iface.launch()