Spaces:
Configuration error
Configuration error
Commit
·
a1586fe
1
Parent(s):
5b8252c
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,41 +14,3 @@ description = """
|
|
| 14 |
</center>
|
| 15 |
</p>
|
| 16 |
"""
|
| 17 |
-
from utils.hparams import hparams
|
| 18 |
-
from preprocessing.data_gen_utils import get_pitch_parselmouth,get_pitch_crepe
|
| 19 |
-
import numpy as np
|
| 20 |
-
import matplotlib.pyplot as plt
|
| 21 |
-
import IPython.display as ipd
|
| 22 |
-
import utils
|
| 23 |
-
import librosa
|
| 24 |
-
import torchcrepe
|
| 25 |
-
from infer import *
|
| 26 |
-
import logging
|
| 27 |
-
from infer_tools.infer_tool import *
|
| 28 |
-
##EDIT FOR CPU
|
| 29 |
-
|
| 30 |
-
# Open the file and read it into a string
|
| 31 |
-
with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py") as f:
|
| 32 |
-
text = f.read()
|
| 33 |
-
|
| 34 |
-
# Replace the original line with the new line
|
| 35 |
-
text = text.replace("def load(f, map_location=None, pickle_module=pickle, **pickle_load_args):", "def load(f, map_location='cpu', pickle_module=pickle, **pickle_load_args):")
|
| 36 |
-
# Save the modified string to the original file
|
| 37 |
-
with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py", "w") as f:
|
| 38 |
-
f.write(text)
|
| 39 |
-
|
| 40 |
-
print("Replaced")
|
| 41 |
-
with open("/home/user/.local/lib/python3.8/site-packages/torch/serialization.py") as f:
|
| 42 |
-
text = f.read()
|
| 43 |
-
print(text)
|
| 44 |
-
|
| 45 |
-
############
|
| 46 |
-
logging.getLogger('numba').setLevel(logging.WARNING)
|
| 47 |
-
|
| 48 |
-
# 工程文件夹名,训练时用的那个
|
| 49 |
-
project_name = "Unnamed"
|
| 50 |
-
model_path = f'./checkpoints/Unnamed/model_ckpt_steps_192000.ckpt'
|
| 51 |
-
config_path=f'./checkpoints/Unnamed/config.yaml'
|
| 52 |
-
hubert_gpu=False
|
| 53 |
-
svc_model = Svc(project_name,config_path,hubert_gpu, model_path)
|
| 54 |
-
print('model loaded')
|
|
|
|
| 14 |
</center>
|
| 15 |
</p>
|
| 16 |
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|