Martijn Bartelds commited on
Commit
6425ecd
·
1 Parent(s): 300bd05

Update app

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,13 +21,13 @@ gos_text2speech = Text2Speech.from_pretrained(
21
  def inference(text,lang):
22
  with torch.no_grad():
23
  if lang == "Hoogelaandsters":
24
- wav = gos_text2speech(text, sids=np.array([1]))["wav"]
25
  scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
26
  if lang == "Oldambsters":
27
- wav = gos_text2speech(text, sids=np.array([2]))["wav"]
28
  scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
29
  if lang == "Westerkertaaiers":
30
- wav = gos_text2speech(text, sids=np.array([3]))["wav"]
31
  scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
32
 
33
  return "out.wav", "out.wav"
 
21
  def inference(text,lang):
22
  with torch.no_grad():
23
  if lang == "Hoogelaandsters":
24
+ wav = gos_text2speech(text.lower(), sids=np.array([1]))["wav"]
25
  scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
26
  if lang == "Oldambsters":
27
+ wav = gos_text2speech(text.lower(), sids=np.array([2]))["wav"]
28
  scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
29
  if lang == "Westerkertaaiers":
30
+ wav = gos_text2speech(text.lower(), sids=np.array([3]))["wav"]
31
  scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
32
 
33
  return "out.wav", "out.wav"