update 2 app.py

app.py

@@ -20,15 +20,19 @@ from accentor import accentification, stress_replace_and_shift
 
 app = FastAPI(docs_url=None, redoc_url=None)
 
-# Set environment variable for Hugging Face cache directory
+# Set environment variable for Hugging Face cache directory and Stanza resources directory
 os.environ["HF_HOME"] = "/app/.cache"
-
 os.environ["STANZA_RESOURCES_DIR"] = "/app/stanza_resources"
 os.environ["MODEL_DIR"] = "/app/model"
 
 # Create necessary directories if they don't exist
-os.makedirs("/app/stanza_resources", exist_ok=True)
-os.makedirs("/app/model", exist_ok=True)
+os.makedirs(os.environ["STANZA_RESOURCES_DIR"], exist_ok=True)
+os.makedirs(os.environ["MODEL_DIR"], exist_ok=True)
+
+# Download the Ukrainian models for Stanza
+import stanza
+
+stanza.download('uk', model_dir=os.environ["STANZA_RESOURCES_DIR"])
 
 tts_kwargs = {
     "speaker_name": "uk",
@@ -97,7 +101,7 @@ def trim_memory():
 
 def init_models():
     models = {}
-    model_path = hf_hub_download("theodotus/tts-vits-lada-uk", "model.pt", cache_dir="/app/model")
+    model_path = hf_hub_download("theodotus/tts-vits-lada-uk", "model.pt", cache_dir=os.environ["MODEL_DIR"])
     importer = package.PackageImporter(model_path)
     models["lada"] = importer.load_pickle("tts_models", "model")
     return models
@@ -111,7 +115,7 @@ async def tts(request: str, api_data: Auth = Depends(check_api_token)):
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as wav_fp:
         with no_grad():
             wav_data = synt.tts(plussed_text, **tts_kwargs)
-            synt.save_wav(wav_data, wav_fp)
+            synt.save_wav(wav_data, wav_fp.name)
 
         threading.Thread(target=delete_file_after_delay, args=(wav_fp.name, 300)).start()
         return JSONResponse(content={"audio_url": f"https://pro100sata-xche-audio.hf.space/download_audio?audio_path={wav_fp.name}"})

requirements.txt

@@ -8,4 +8,5 @@ uvicorn
 pydub
 passlib
 ctranslate2
-sentencepiece
+sentencepiece
+stanza