Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ logging.basicConfig(
|
|
23 |
)
|
24 |
logger = logging.getLogger(__name__)
|
25 |
|
26 |
-
# ===== تحميل النماذج =====
|
27 |
try:
|
28 |
# 1. نموذج التعرف على الكلام
|
29 |
asr_pipeline = pipeline(
|
@@ -32,9 +32,9 @@ try:
|
|
32 |
token=os.getenv("HF_TOKEN")
|
33 |
)
|
34 |
|
35 |
-
# 2. نموذج توليف الصوت الأنثوي (فيسبوك)
|
36 |
tts_tokenizer = AutoTokenizer.from_pretrained(
|
37 |
-
"facebook/mms-tts-arb", #
|
38 |
token=os.getenv("HF_TOKEN")
|
39 |
)
|
40 |
tts_model = VitsModel.from_pretrained(
|
@@ -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=1) # الصوت الأنثوي
|
104 |
waveform = output.waveform[0].numpy()
|
105 |
sf.write("bot_response.wav", waveform, tts_model.config.sampling_rate)
|
106 |
except Exception as e:
|
|
|
23 |
)
|
24 |
logger = logging.getLogger(__name__)
|
25 |
|
26 |
+
# ===== تحميل النماذج البديلة =====
|
27 |
try:
|
28 |
# 1. نموذج التعرف على الكلام
|
29 |
asr_pipeline = pipeline(
|
|
|
32 |
token=os.getenv("HF_TOKEN")
|
33 |
)
|
34 |
|
35 |
+
# 2. نموذج توليف الصوت الأنثوي (فيسبوك MMS)
|
36 |
tts_tokenizer = AutoTokenizer.from_pretrained(
|
37 |
+
"facebook/mms-tts-arb", # نموذج بديل مدعوم
|
38 |
token=os.getenv("HF_TOKEN")
|
39 |
)
|
40 |
tts_model = VitsModel.from_pretrained(
|
|
|
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:
|