Update app.py

app.py

@@ -7,33 +7,33 @@ from pydantic import BaseModel
 from typing import List, Optional
 from pathlib import Path
 
-
-# MODEL_DIR = "my_model"
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 from huggingface_hub import hf_hub_download
+
 # ------------------------
 # Download model files from Hugging Face if not present
 # ------------------------
 MODEL_DIR = "my_model"
+
 config_path = hf_hub_download(
-    repo_id="MariaKaiser/EGTTS_finetuned",
-    filename="config.json",
+    repo_id="MariaKaiser/egtts_finetuned_with_vocab",
+    filename="my_model/config.json",
     cache_dir=MODEL_DIR
 )
 
 vocab_path = hf_hub_download(
-    repo_id="MariaKaiser/EGTTS_finetuned",
-    filename="vocab.json",
+    repo_id="MariaKaiser/egtts_finetuned_with_vocab",
+    filename="my_model/vocab.json",
     cache_dir=MODEL_DIR
 )
 
 model_path = hf_hub_download(
-    repo_id="MariaKaiser/EGTTS_finetuned",
-    filename="model.pth",
+    repo_id="MariaKaiser/egtts_finetuned_with_vocab",
+    filename="my_model/model.pth",
     cache_dir=MODEL_DIR
 )
 
@@ -330,17 +330,17 @@ async def process_story(story: StoryCreationDTO):
     return response
     
 
-
-# async def tts_endpoint(
-#     text: str = Form(...),
-#     audio_file: UploadFile = File(...)
-# ):
-#     file_path = os.path.join(OUTPUT_DIR, audio_file.filename)
-#     with open(file_path, "wb") as f:
-#         f.write(await audio_file.read())
-
-#     output_wav = tts_arabic(text, file_path)
-#     return FileResponse(output_wav, media_type="audio/wav", filename="output.wav")
+@app.post("/tts_test/")
+async def tts_endpoint(
+    text: str = Form(...),
+    audio_file: UploadFile = File(...)
+):
+    file_path = os.path.join(OUTPUT_DIR, audio_file.filename)
+    with open(file_path, "wb") as f:
+        f.write(await audio_file.read())
+
+    output_wav = tts_arabic(text, file_path)
+    return FileResponse(output_wav, media_type="audio/wav", filename="output.wav")
 
 import uvicorn
 uvicorn.run(app, host="0.0.0.0", port=7860)