Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -34,11 +34,11 @@ try:
|
|
34 |
|
35 |
# 2. نموذج توليف الصوت الأنثوي (الاسم الصحيح)
|
36 |
tts_tokenizer = AutoTokenizer.from_pretrained(
|
37 |
-
"
|
38 |
token=os.getenv("HF_TOKEN")
|
39 |
)
|
40 |
tts_model = VitsModel.from_pretrained(
|
41 |
-
"
|
42 |
token=os.getenv("HF_TOKEN")
|
43 |
)
|
44 |
|
@@ -100,7 +100,7 @@ async def text_to_speech(text: str) -> None:
|
|
100 |
try:
|
101 |
inputs = tts_tokenizer(text, return_tensors="pt")
|
102 |
with torch.no_grad():
|
103 |
-
output = tts_model(**inputs, speaker_id=
|
104 |
waveform = output.waveform[0].numpy()
|
105 |
sf.write("bot_response.wav", waveform, tts_model.config.sampling_rate)
|
106 |
except Exception as e:
|
|
|
34 |
|
35 |
# 2. نموذج توليف الصوت الأنثوي (الاسم الصحيح)
|
36 |
tts_tokenizer = AutoTokenizer.from_pretrained(
|
37 |
+
"facebook/mms-tts-ara", # تم تغيير النموذج إلى فيسبوك MMS
|
38 |
token=os.getenv("HF_TOKEN")
|
39 |
)
|
40 |
tts_model = VitsModel.from_pretrained(
|
41 |
+
"facebook/mms-tts-ara",
|
42 |
token=os.getenv("HF_TOKEN")
|
43 |
)
|
44 |
|
|
|
100 |
try:
|
101 |
inputs = tts_tokenizer(text, return_tensors="pt")
|
102 |
with torch.no_grad():
|
103 |
+
output = tts_model(**inputs, speaker_id=1) # اختيار الصوت الأنثوي
|
104 |
waveform = output.waveform[0].numpy()
|
105 |
sf.write("bot_response.wav", waveform, tts_model.config.sampling_rate)
|
106 |
except Exception as e:
|