ASG Models commited on
Commit
1318041
·
verified ·
1 Parent(s): 2a84d5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def genrate_speech(text,name_model):
25
  model=get_model(name_model)
26
  with torch.no_grad():
27
  wav=model(
28
- input_ids= input_ids.input_ids.to(device),
29
- attention_mask=input_ids.attention_mask.to(device),
30
  speaker_id=0
31
  ).waveform.cpu().numpy().reshape(-1)
32
  return model.config.sampling_rate,wav
 
25
  model=get_model(name_model)
26
  with torch.no_grad():
27
  wav=model(
28
+ input_ids= inputs.input_ids.to(device),
29
+ attention_mask=inputs.attention_mask.to(device),
30
  speaker_id=0
31
  ).waveform.cpu().numpy().reshape(-1)
32
  return model.config.sampling_rate,wav