Update app.py
Browse files
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)
|