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)
|