add is disk full check

interfaces/ontolisst.py

@@ -15,6 +15,8 @@ languages = [
 
 from label_dicts import ONTOLISST_LABEL_NAMES
 
+from utils import is_disk_full
+
 # --- DEBUG ---
 import shutil
 
@@ -79,6 +81,11 @@ def predict(text, model_id, tokenizer_id):
 def predict_cap(text, language):
     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(