2506minecraft commited on
Commit
e7e0232
·
verified ·
1 Parent(s): fbc664b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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: