Kangarroar's picture
Update app.py
642e2a7
raw
history blame
No virus
1.18 kB
title = "DIFF-SVC"
description = """
<p>
<body style="background-color: #18181a; color: white;"></body>
<center>
<h1>DIFF-SVC Inference Cloud</h1>
This is a Cloud Inference where you can render your models with your wav files
<p>Enter a link:</p>
<input type="text" id="link-input"/>
<p>Upload a WAV file:</p> <input type="file" id="wav-input" accept=".wav"/>
<button id="render-button">Render</button>
<p>Diff-SVC prediction:</p>
<p id="prediction-output"></p>
</center>
</p>
"""
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 *
############
logging.getLogger('numba').setLevel(logging.WARNING)
# 工程文件夹名,训练时用的那个
project_name = "Unnamed"
model_path = f'./checkpoints/Unnamed/model_ckpt_steps_192000.ckpt'
config_path=f'./checkpoints/Unnamed/config.yaml'
hubert_gpu=False
svc_model = Svc(project_name,config_path,hubert_gpu, model_path)
print('model loaded')