add is disk full check

interfaces/illframes.py

@@ -10,6 +10,8 @@ from huggingface_hub import HfApi
 
 from label_dicts import ILLFRAMES_MIGRATION_LABEL_NAMES, ILLFRAMES_COVID_LABEL_NAMES, ILLFRAMES_WAR_LABEL_NAMES
 
+from utils import is_disk_full
+
 HF_TOKEN = os.environ["hf_read"]
 
 languages = [
@@ -99,6 +101,10 @@ def predict_illframes(text, language, domain):
     elif domain == "war":
         label_names = ILLFRAMES_WAR_LABEL_NAMES
 
+    if is_disk_full():
+        os.system('rm -rf /data/models*')
+        os.system('rm -r ~/.cache/huggingface/hub')
+
     return predict(text, model_id, tokenizer_id, label_names)
 
 demo = gr.Interface(