| import infer_web |
| import wget |
| import os |
| import scipy.io.wavfile as wavfile |
| from utils import model |
| import validators |
| from myutils import delete_files |
|
|
| class Inference: |
| |
| inference_cont = 0 |
| |
| def __init__( |
| self, |
| model_name=None, |
| source_audio_path=None, |
| output_file_name=None, |
| feature_index_path="", |
| f0_file=None, |
| speaker_id=0, |
| transposition=-2, |
| f0_method="harvest", |
| crepe_hop_length=160, |
| harvest_median_filter=3, |
| resample=0, |
| mix=1, |
| feature_ratio=0.78, |
| protection_amnt=0.33, |
| protect1=False |
| ): |
| Inference.inference_cont += 1 |
| self._model_name = model_name |
| self._source_audio_path = source_audio_path |
| self._output_file_name = output_file_name |
| self._feature_index_path = feature_index_path |
| self._f0_file = f0_file |
| self._speaker_id = speaker_id |
| self._transposition = transposition |
| self._f0_method = f0_method |
| self._crepe_hop_length = crepe_hop_length |
| self._harvest_median_filter = harvest_median_filter |
| self._resample = resample |
| self._mix = mix |
| self._feature_ratio = feature_ratio |
| self._protection_amnt = protection_amnt |
| self._protect1 = protect1 |
| self._id = Inference.inference_cont |
|
|
| if not os.path.exists("./hubert_base.pt"): |
| wget.download( |
| "https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/hubert_base.pt", out="./hubert_base.pt") |
| |
| if not os.path.exists("./rmvpe.pt"): |
| wget.download( |
| "https://huggingface.co/lj1995/VoiceConversionWebUI/resolve/main/rmvpe.pt", out="./rmvpe.pt" |
| ) |
| |
| @property |
| def id(self): |
| return self._id |
|
|
| @id.setter |
| def id(self, id): |
| self._id = id |
| |
| @property |
| def audio(self): |
| return self._audio |
|
|
| @audio.setter |
| def audio_file(self, audio): |
| self._audio_file = audio |
|
|
| @property |
| def model_name(self): |
| return self._model_name |
|
|
| @model_name.setter |
| def model_name(self, model_name): |
| self._model_name = model_name |
|
|
| @property |
| def source_audio_path(self): |
| return self._source_audio_path |
|
|
| @source_audio_path.setter |
| def source_audio_path(self, source_audio_path): |
| if not self._output_file_name: |
| self._output_file_name = os.path.join("./audio-outputs", os.path.basename(source_audio_path)) |
| self._source_audio_path = source_audio_path |
|
|
| @property |
| def output_file_name(self): |
| return self._output_file_name |
|
|
| @output_file_name.setter |
| def output_file_name(self, output_file_name): |
| self._output_file_name = output_file_name |
|
|
| @property |
| def feature_index_path(self): |
| return self._feature_index_path |
|
|
| @feature_index_path.setter |
| def feature_index_path(self, feature_index_path): |
| self._feature_index_path = feature_index_path |
|
|
| @property |
| def f0_file(self): |
| return self._f0_file |
|
|
| @f0_file.setter |
| def f0_file(self, f0_file): |
| self._f0_file = f0_file |
|
|
| @property |
| def speaker_id(self): |
| return self._speaker_id |
|
|
| @speaker_id.setter |
| def speaker_id(self, speaker_id): |
| self._speaker_id = speaker_id |
|
|
| @property |
| def transposition(self): |
| return self._transposition |
|
|
| @transposition.setter |
| def transposition(self, transposition): |
| self._transposition = transposition |
|
|
| @property |
| def f0_method(self): |
| return self._f0_method |
|
|
| @f0_method.setter |
| def f0_method(self, f0_method): |
| self._f0_method = f0_method |
|
|
| @property |
| def crepe_hop_length(self): |
| return self._crepe_hop_length |
|
|
| @crepe_hop_length.setter |
| def crepe_hop_length(self, crepe_hop_length): |
| self._crepe_hop_length = crepe_hop_length |
|
|
| @property |
| def harvest_median_filter(self): |
| return self._harvest_median_filter |
|
|
| @crepe_hop_length.setter |
| def harvest_median_filter(self, harvest_median_filter): |
| self._harvest_median_filter = harvest_median_filter |
|
|
| @property |
| def resample(self): |
| return self._resample |
|
|
| @resample.setter |
| def resample(self, resample): |
| self._resample = resample |
|
|
| @property |
| def mix(self): |
| return self._mix |
|
|
| @mix.setter |
| def mix(self, mix): |
| self._mix = mix |
|
|
| @property |
| def feature_ratio(self): |
| return self._feature_ratio |
|
|
| @feature_ratio.setter |
| def feature_ratio(self, feature_ratio): |
| self._feature_ratio = feature_ratio |
|
|
| @property |
| def protection_amnt(self): |
| return self._protection_amnt |
|
|
| @protection_amnt.setter |
| def protection_amnt(self, protection_amnt): |
| self._protection_amnt = protection_amnt |
|
|
| @property |
| def protect1(self): |
| return self._protect1 |
|
|
| @protect1.setter |
| def protect1(self, protect1): |
| self._protect1 = protect1 |
|
|
| def run(self): |
| current_dir = os.getcwd() |
| modelname = model.model_downloader(self._model_name, "./zips/", "./weights/") |
| |
| model_info = model.get_model(os.path.join(current_dir, 'weights') , modelname) |
| if not model_info: |
| return "No se encontrado un modelo valido, verifica el contenido del enlace e intentalo más tarde." |
|
|
| if not model_info.get('pth'): |
| return "No se encontrado un modelo valido, verifica el contenido del enlace e intentalo más tarde." |
| |
| index = model_info.get('index', '') |
| pth = model_info.get('pth', None) |
| |
| infer_web.get_vc(pth) |
| |
| conversion_data = infer_web.vc_single( |
| self.speaker_id, |
| self.source_audio_path, |
| self.source_audio_path, |
| self.transposition, |
| self.f0_file, |
| self.f0_method, |
| index, |
| index, |
| self.feature_ratio, |
| self.harvest_median_filter, |
| self.resample, |
| self.mix, |
| self.protection_amnt, |
| self.crepe_hop_length, |
| ) |
| |
| if "Success." in conversion_data[0]: |
| wavfile.write( |
| "%s/%s" % ("audio-outputs",os.path.basename(self._output_file_name)), |
| conversion_data[1][0], |
| conversion_data[1][1], |
| ) |
| return({ |
| "success": True, |
| "file": self._output_file_name |
| }) |
| else: |
| return({ |
| "success": False, |
| "file": self._output_file_name |
| }) |
| |
