Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -141,9 +141,9 @@ TXT="""السلام عليكم ورحمة الله وبركاتة يا هلا
|
|
| 141 |
@spaces.GPU
|
| 142 |
def modelspeech(text=TXT,name_model="wasmdashai/vits-ar-sa-huba-v2",speaking_rate=16000):
|
| 143 |
|
| 144 |
-
|
| 145 |
inputs = tokenizer(text, return_tensors="pt")
|
| 146 |
-
|
| 147 |
model.speaking_rate=speaking_rate
|
| 148 |
with torch.no_grad():
|
| 149 |
wavs=_inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= None,is_streaming=True)
|
|
|
|
| 141 |
@spaces.GPU
|
| 142 |
def modelspeech(text=TXT,name_model="wasmdashai/vits-ar-sa-huba-v2",speaking_rate=16000):
|
| 143 |
|
| 144 |
+
model,tokenizer=get_model(name_model)
|
| 145 |
inputs = tokenizer(text, return_tensors="pt")
|
| 146 |
+
|
| 147 |
model.speaking_rate=speaking_rate
|
| 148 |
with torch.no_grad():
|
| 149 |
wavs=_inference_forward_stream(model,input_ids=inputs.input_ids,attention_mask=inputs.attention_mask,speaker_embeddings= None,is_streaming=True)
|