Update app.py

app.py

@@ -6,36 +6,39 @@ from flask import Flask, request, jsonify, send_file
 from flask_cors import CORS
 from transformers import VitsModel, AutoTokenizer
 
-# Set environment variables for Hugging Face cache (prevents permission issues)
-os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface_cache"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
+
+# Set ALL cache directories to /tmp (writable in Hugging Face Spaces)
 os.environ["HF_HOME"] = "/tmp/hf_home"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
+os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/huggingface_cache"
 os.environ["TORCH_HOME"] = "/tmp/torch_home"
 
+
 app = Flask(__name__)
 CORS(app)  # Allow external requests
 
-# Language models (Hugging Face)
+# Model paths for different languages (Hugging Face Hub)
 MODELS = {
     "kapampangan": "facebook/mms-tts-pam",
     "tagalog": "facebook/mms-tts-tgl",
     "english": "facebook/mms-tts-eng"
 }
 
-# Model storage
 loaded_models = {}
 loaded_processors = {}
 
-# Load models and processors
-for lang, model_path in MODELS.items():
+for lang, path in MODELS.items():
     try:
-        print(f"🔄 Loading {lang} model: {model_path}...")
-        loaded_models[lang] = VitsModel.from_pretrained(model_path)
-        loaded_processors[lang] = AutoTokenizer.from_pretrained(model_path)
+        print(f"🔄 Loading {lang} model: {path}...")
+
+        # Force models to save in /tmp
+        loaded_models[lang] = VitsModel.from_pretrained(path, cache_dir="/tmp/huggingface_cache")
+        loaded_processors[lang] = AutoTokenizer.from_pretrained(path, cache_dir="/tmp/huggingface_cache")
+
         print(f"✅ {lang.capitalize()} model loaded successfully!")
     except Exception as e:
-        print(f"❌ Error loading {lang} model: {e}")
-        loaded_models[lang] = None
+        print(f"❌ Error loading {lang} model: {str(e)}")
+        loaded_models[lang] = None  # Mark as unavailable
         loaded_processors[lang] = None
 
 # Constants