bark-voice-cloning-polish-HuBERT-quantizer

Runtime error

Hobis commited on May 28, 2023

Commit

001ec5c

1 Parent(s): b80ddf0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,8 @@ from encodec.utils import convert_audio
 hubert_model = CustomHubert(checkpoint_path='hubert.pt')
 model = EncodecModel.encodec_model_24khz()
 model.set_target_bandwidth(6.0)
-tokenizer_model = CustomTokenizer.load_from_checkpoint('polish-HuBERT-quantizer_8_epoch.pth', map_location=torch.device('cpu'))
 def process_audio(in_file):
     wav, sr = torchaudio.load(in_file.name)

 hubert_model = CustomHubert(checkpoint_path='hubert.pt')
 model = EncodecModel.encodec_model_24khz()
 model.set_target_bandwidth(6.0)
+tokenizer = CustomTokenizer.load_from_checkpoint('polish-HuBERT-quantizer_8_epoch.pth', map_location=torch.device('cpu'))
 def process_audio(in_file):
     wav, sr = torchaudio.load(in_file.name)