Use cached snapshot download for model files

app.py

@@ -16,7 +16,6 @@ from supertonic_mlx import SupertonicMLX
 os.environ.setdefault("HF_HUB_ENABLE_HF_TRANSFER", "1")
 
 MODEL_ID = os.environ.get("SUPERTONIC_MLX_MODEL", "mlx-community/supertonic-3")
-MODEL_DIR = os.environ.get("SUPERTONIC_MLX_MODEL_DIR", "/models/supertonic-3")
 LANGUAGES = [
     "en", "ko", "ja", "ar", "bg", "cs", "da", "de", "el", "es", "et", "fi",
     "fr", "hi", "hr", "hu", "id", "it", "lt", "lv", "nl", "pl", "pt", "ro",
@@ -27,21 +26,18 @@ VOICES = ["M1", "M2", "M3", "M4", "M5", "F1", "F2", "F3", "F4", "F5"]
 
 @lru_cache(maxsize=1)
 def load_model() -> SupertonicMLX:
-    if os.path.exists(os.path.join(MODEL_DIR, "tts.json")):
-        model_dir = MODEL_DIR
-    else:
-        model_dir = snapshot_download(
-            repo_id=MODEL_ID,
-            repo_type="model",
-            allow_patterns=[
-                "graphs/*.json",
-                "weights/*.npz",
-                "voice_styles/*.json",
-                "tts.json",
-                "unicode_indexer.json",
-                "mlx_manifest.json",
-            ],
-        )
+    model_dir = snapshot_download(
+        repo_id=MODEL_ID,
+        repo_type="model",
+        allow_patterns=[
+            "graphs/*.json",
+            "weights/*.npz",
+            "voice_styles/*.json",
+            "tts.json",
+            "unicode_indexer.json",
+            "mlx_manifest.json",
+        ],
+    )
     return SupertonicMLX.from_pretrained(model_dir)