add is disk full check

interfaces/sentiment.py

@@ -9,6 +9,8 @@ from huggingface_hub import HfApi
 
 from label_dicts import MANIFESTO_LABEL_NAMES
 
+from utils import is_disk_full
+
 HF_TOKEN = os.environ["hf_read"]
 
 languages = [
@@ -56,6 +58,11 @@ def predict(text, model_id, tokenizer_id):
 def predict_cap(text, language, domain):
     model_id = build_huggingface_path(language)
     tokenizer_id = "xlm-roberta-large"
+
+    if is_disk_full():
+        os.system('rm -rf /data/models*')
+        os.system('rm -r ~/.cache/huggingface/hub')
+    
     return predict(text, model_id, tokenizer_id)
 
 demo = gr.Interface(