from utils.hparams import hparams
from preprocessing.data_gen_utils import get_pitch_parselmouth,get_pitch_crepe
import numpy as np
import matplotlib.pyplot as plt
import IPython.display as ipd
import utils
import librosa
import torchcrepe
from infer import *
import logging
from infer_tools.infer_tool import *
import gradio as gr
import transformers
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# define the input interface for the app
inputs = gr.inputs.Textbox(lines=1, label="Enter a link:")
inputs += gr.inputs.Audio(label="Upload a WAV file:")

# define the output interface for the app
outputs = gr.outputs.Textbox(lines=1, label="Diff-SVC prediction:")

# define the app itself
app = gr.Interface(model, inputs, outputs, title="DIFF-SVC Inference Cloud")

# run the app
app.launch()