farid678 commited on
Commit
a766e2a
·
verified ·
1 Parent(s): 2f8b494

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -3
app.py CHANGED
@@ -29,15 +29,27 @@ def tts_generate(text):
29
  if not text.strip():
30
  return None
31
 
 
32
  output = tts_pipe(text)
33
 
34
- audio = np.asarray(output["audio"], dtype=np.float32)
35
- sr = output["sampling_rate"]
 
36
 
 
 
 
 
 
 
 
 
 
37
  os.makedirs("outputs", exist_ok=True)
38
  out_path = f"outputs/{uuid.uuid4().hex}.wav"
39
 
40
- write(out_path, sr, audio)
 
41
 
42
  return out_path
43
 
@@ -67,3 +79,4 @@ demo = gr.Interface(
67
 
68
  if __name__ == "__main__":
69
  demo.launch()
 
 
29
  if not text.strip():
30
  return None
31
 
32
+ # اجرای مدل TTS
33
  output = tts_pipe(text)
34
 
35
+ # بررسی خروجی
36
+ if "audio" not in output:
37
+ raise ValueError("TTS pipeline did not return audio")
38
 
39
+ audio = np.array(output["audio"], dtype=np.float32)
40
+
41
+ # بررسی sampling_rate و مقدار پیش‌فرض در صورت نبود
42
+ sr = output.get("sampling_rate", 22050)
43
+
44
+ # تبدیل float32 به int16 برای scipy
45
+ audio_int16 = (audio * 32767).astype(np.int16)
46
+
47
+ # ساخت پوشه خروجی
48
  os.makedirs("outputs", exist_ok=True)
49
  out_path = f"outputs/{uuid.uuid4().hex}.wav"
50
 
51
+ # ذخیره فایل WAV
52
+ write(out_path, sr, audio_int16)
53
 
54
  return out_path
55
 
 
79
 
80
  if __name__ == "__main__":
81
  demo.launch()
82
+