Switch to hf model

.gitattributes

@@ -33,15 +33,10 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-Utils/JDC/bst.t7 filter=lfs diff=lfs merge=lfs -text
-Utils/PLBERT/step_410000.t7 filter=lfs diff=lfs merge=lfs -text
-epoch_2nd_00027_filatov_whisp_cont.pth filter=lfs diff=lfs merge=lfs -text
-Utils/ASR/epoch_00100.pth filter=lfs diff=lfs merge=lfs -text
 weights/plbert.bin filter=lfs diff=lfs merge=lfs -text
 weights/asr.bin filter=lfs diff=lfs merge=lfs -text
 weights/filatov.bin filter=lfs diff=lfs merge=lfs -text
 weights/jdc.bin filter=lfs diff=lfs merge=lfs -text
-weights/multi.bin filter=lfs diff=lfs merge=lfs -text
 voices/Олег[[:space:]]Лепенець.wav filter=lfs diff=lfs merge=lfs -text
 voices/Слава[[:space:]]Красовська.wav filter=lfs diff=lfs merge=lfs -text
 voices/Гаська[[:space:]]Шиян.wav filter=lfs diff=lfs merge=lfs -text

infer.py

@@ -10,6 +10,7 @@ import numpy as np
 np.random.seed(0)
 import librosa
 from copy import deepcopy
+from huggingface_hub import hf_hub_download
 
 import spaces
 import yaml
@@ -80,8 +81,10 @@ model_single = build_model(recursive_munch(config['model_params_single']), text_
 model_multi = build_model(recursive_munch(config['model_params_multi']), deepcopy(text_aligner), deepcopy(pitch_extractor), deepcopy(plbert))
 
 
+multi_path = hf_hub_download(repo_id='patriotyk/styletts2_ukrainian_multispeaker', filename="pytorch_model.bin")
+params_multi = torch.load(multi_path, map_location='cpu')
+
 
-params_multi = torch.load('weights/multi.bin', map_location='cpu')
 params_single = torch.load('weights/filatov.bin', map_location='cpu')