Update app.py

app.py

@@ -1,9 +1,17 @@
+
+import os
+
+# Set ALL cache directories before importing any model libraries
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
+os.environ["HF_HOME"] = "/tmp/hf_home"
+os.environ["TORCH_HOME"] = "/tmp/torch_home"
+
 from flask import Flask, request, jsonify, send_file
 from flask_cors import CORS
 import torch
 import torchaudio
 import soundfile as sf
-import os
 from transformers import VitsModel, AutoTokenizer
 
 app = Flask(__name__)
@@ -24,10 +32,13 @@ loaded_models = {}
 loaded_processors = {}
 
 for lang, path in MODELS.items():
-    print(f"Loading {lang} model from Hugging Face: {path}...")
-    loaded_models[lang] = VitsModel.from_pretrained(path)
-    loaded_processors[lang] = AutoTokenizer.from_pretrained(path)
-    print(f"{lang.capitalize()} model loaded successfully!")
+    try:
+        print(f"Loading {lang} model from Hugging Face: {path}...")
+        loaded_models[lang] = VitsModel.from_pretrained(path)
+        loaded_processors[lang] = AutoTokenizer.from_pretrained(path)
+        print(f"{lang.capitalize()} model loaded successfully!")
+    except Exception as e:
+        print(f"Error loading {lang} model: {str(e)}")
 
 SAMPLE_RATE = 16000  # Default sample rate
 OUTPUT_DIR = "/tmp/"  # Use /tmp for Hugging Face Spaces (limited storage)