Update app.py

app.py

@@ -4,9 +4,24 @@ import numpy as np
 import scipy.io.wavfile
 from transformers import VitsModel, AutoTokenizer
 import re
+import os
+import shutil
+
+# --- ADDITIONAL CODE TO FIX STORAGE ERROR ---
+# This clears the Hugging Face cache to free up the 50GB disk space limit
+cache_path = os.path.expanduser("~/.cache/huggingface")
+if os.path.exists(cache_path):
+    try:
+        shutil.rmtree(cache_path)
+        os.makedirs(cache_path)
+        print("Cache cleared successfully to free up storage.")
+    except Exception as e:
+        print(f"Note: Could not clear cache: {e}")
+# ---------------------------------------------
 
 # Load model and tokenizer
-model = VitsModel.from_pretrained("Somali-tts/somali_tts_model")
+# Added low_cpu_mem_usage=True to help with Space stability
+model = VitsModel.from_pretrained("Somali-tts/somali_tts_model", low_cpu_mem_usage=True)
 tokenizer = AutoTokenizer.from_pretrained("saleolow/somali-mms-tts")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
@@ -203,4 +218,4 @@ gr.Interface(
     outputs=gr.Audio(label="Codka TTS"),
     title="Somali TTS",
     description="Ku qor qoraal Soomaaliyeed si aad u maqasho cod dabiici ah. Qoraalka ha ka badnaan 2 daqiiqo per jumlad."
-).launch()
+).launch()