import gradio as gr
from inference import InferencePipeline

# Initialize your model or inference pipeline
i = InferencePipeline()

# Define your Gradio interface
demo = gr.Interface(
    fn=i.voice_conversion,  # The function to convert voice
    inputs=gr.Audio(label="Record or upload your voice"),  # Define the input as audio
    outputs=gr.Audio(label="Converted Voice"),  # Define the output as audio
    title="Voice Conversion Demo",
    description="Voice Conversion: Transform the input voice to a target voice.",
    allow_flagging="never"
)

if __name__ == "__main__":
    demo.launch()