Alidr79 commited on
Commit
2b283a1
·
verified ·
1 Parent(s): c4eea5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -22,8 +22,8 @@ set_seed(1)
22
  # Load model directly
23
  from transformers import AutoProcessor, AutoModelForTextToSpectrogram
24
 
25
- processor = AutoProcessor.from_pretrained("Alidr79/speecht5_v2_best")
26
- model = AutoModelForTextToSpectrogram.from_pretrained("Alidr79/speecht5_v2_best")
27
 
28
 
29
  from speechbrain.inference.classifiers import EncoderClassifier
@@ -82,11 +82,11 @@ def tts_fn(slider_value, input_text):
82
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
83
 
84
  phonemes = PersianG2Pconverter.transliterate(input_text, tidy = False, secret = True)
85
- text = "</s>"
86
- for i in phonemes.replace(' .', '').split(" "):
87
- text += i + " <pad> "
88
 
89
- text += "</s>"
90
 
91
  print("sentence:", input_text)
92
  print("sentence phonemes:", text)
 
22
  # Load model directly
23
  from transformers import AutoProcessor, AutoModelForTextToSpectrogram
24
 
25
+ processor = AutoProcessor.from_pretrained("Alidr79/speecht5_v3_youtube")
26
+ model = AutoModelForTextToSpectrogram.from_pretrained("Alidr79/speecht5_v3_youtube")
27
 
28
 
29
  from speechbrain.inference.classifiers import EncoderClassifier
 
82
  speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
83
 
84
  phonemes = PersianG2Pconverter.transliterate(input_text, tidy = False, secret = True)
85
+ # text = "</s>"
86
+ # for i in phonemes.replace(' .', '').split(" "):
87
+ # text += i + " <pad> "
88
 
89
+ text = phonemes
90
 
91
  print("sentence:", input_text)
92
  print("sentence phonemes:", text)