Spaces:
Sleeping
Sleeping
ASG Models
commited on
Update app.py
Browse files
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):
|