import gradio as gr
import hopsworks
def vocal_remove(audio):    
    project = hopsworks.login()
    mr = project.get_model_registry()
    model = mr.get_best_model()
    model_path = model.download()
    model_path_pkl = model_path + "/vocal_model.pth"
    

    
    return audio

iface = gr.Interface(
    fn=vocal_remove, 
    inputs=gr.Audio(source="upload", type="filepath"), 
    outputs="audio",
    title="Vocal Remover",
    description="Removes Vocals from song",
)

iface.launch()