ykirpichev commited on
Commit
720672d
1 Parent(s): 3bfb858

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -38,6 +38,8 @@ def synthesise(text):
38
  with torch.no_grad():
39
  outputs = model_mms(input_ids)
40
  print("mms model", outputs)
 
 
41
  inputs = processor(text=text, return_tensors="pt")
42
  speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
43
  print("speecht5 model", speech)
 
38
  with torch.no_grad():
39
  outputs = model_mms(input_ids)
40
  print("mms model", outputs)
41
+ print(outputs.audio[0])
42
+ return outputs.audio[0].cpu()
43
  inputs = processor(text=text, return_tensors="pt")
44
  speech = model.generate_speech(inputs["input_ids"].to(device), speaker_embeddings.to(device), vocoder=vocoder)
45
  print("speecht5 model", speech)