update app.py

app.py

@@ -1,26 +1,34 @@
+import argparse
+import ctypes
+import gc
 import os
 import sys
-import time
-import threading
 import tempfile
-import ctypes
-import gc
-from fastapi import FastAPI, Depends, HTTPException, Security
+import threading
+import time
+
+import uvicorn
+from fastapi import FastAPI, Depends, HTTPException
 from fastapi.responses import FileResponse, JSONResponse
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
-from pydantic import BaseModel
 from huggingface_hub import hf_hub_download
+from passlib.context import CryptContext
+from pydantic import BaseModel
 from torch import no_grad, package
-import uvicorn
+
 from accentor import accentification, stress_replace_and_shift
-import argparse
-from passlib.context import CryptContext
 
 app = FastAPI(docs_url=None, redoc_url=None)
 
 # Set environment variable for Hugging Face cache directory
 os.environ["HF_HOME"] = "/app/.cache"
 
+os.environ["STANZA_RESOURCES_DIR"] = "/app/stanza_resources"
+
+# Create necessary directories if they don't exist
+os.makedirs("/app/stanza_resources", exist_ok=True)
+os.makedirs("/app/model", exist_ok=True)
+
 tts_kwargs = {
     "speaker_name": "uk",
     "language_name": "uk",
@@ -88,7 +96,7 @@ def trim_memory():
 
 def init_models():
     models = {}
-    model_path = hf_hub_download("theodotus/tts-vits-lada-uk", "model.pt")
+    model_path = hf_hub_download("theodotus/tts-vits-lada-uk", "model.pt", cache_dir="/app/model")
     importer = package.PackageImporter(model_path)
     models["lada"] = importer.load_pickle("tts_models", "model")
     return models