alidw commited on
Commit
fb5ec37
·
verified ·
1 Parent(s): 6417166

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -2
app.py CHANGED
@@ -7,16 +7,19 @@ import io
7
 
8
  app = FastAPI()
9
 
10
- # تحميل موديل MMS-TTS عربي
11
  tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara")
12
  model = VitsModel.from_pretrained("facebook/mms-tts-ara")
13
 
14
  @app.post("/tts")
15
  async def tts_endpoint(text: str):
 
 
 
16
  inputs = tokenizer(text, return_tensors="pt")
17
 
18
  with torch.no_grad():
19
- output = model(**inputs).waveform
20
 
21
  audio = output.squeeze().cpu().numpy()
22
 
@@ -25,3 +28,9 @@ async def tts_endpoint(text: str):
25
  buffer.seek(0)
26
 
27
  return StreamingResponse(buffer, media_type="audio/wav")
 
 
 
 
 
 
 
7
 
8
  app = FastAPI()
9
 
10
+ # تحميل موديل MMS-TTS العربي من Meta
11
  tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara")
12
  model = VitsModel.from_pretrained("facebook/mms-tts-ara")
13
 
14
  @app.post("/tts")
15
  async def tts_endpoint(text: str):
16
+ """
17
+ تستقبل نص عربي وترجع ملف صوت WAV
18
+ """
19
  inputs = tokenizer(text, return_tensors="pt")
20
 
21
  with torch.no_grad():
22
+ output = model(**inputs).waveform # (1, num_samples)
23
 
24
  audio = output.squeeze().cpu().numpy()
25
 
 
28
  buffer.seek(0)
29
 
30
  return StreamingResponse(buffer, media_type="audio/wav")
31
+
32
+
33
+ # مهم جداً: تشغيل السيرفر على البورت 7860 عشان Hugging Face Spaces
34
+ if __name__ == "__main__":
35
+ import uvicorn
36
+ uvicorn.run(app, host="0.0.0.0", port=7860)