ASG Models commited on
Commit
9bba678
·
verified ·
1 Parent(s): eef1809

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -122,6 +122,7 @@ def generate_audio(text, speaker_id=None):
122
  inputs = tokenizer(text, return_tensors="pt")#.input_ids
123
 
124
  speaker_embeddings = None
 
125
  #torch.cuda.empty_cache()
126
  with torch.no_grad():
127
  for chunk in _inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= speaker_embeddings,chunk_size=64):
 
122
  inputs = tokenizer(text, return_tensors="pt")#.input_ids
123
 
124
  speaker_embeddings = None
125
+ model=get_model(name_model)
126
  #torch.cuda.empty_cache()
127
  with torch.no_grad():
128
  for chunk in _inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= speaker_embeddings,chunk_size=64):