remove data folder

interfaces/illframes.py

@@ -66,7 +66,7 @@ def predict(text, model_id, tokenizer_id, label_names):
     # ---
 
     try:
-        model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", token=HF_TOKEN)
+        model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, offload_folder="offload", device_map="auto", token=HF_TOKEN)
     except:
         disk_space = get_disk_space('/data/')
         print("Disk Space Error:")
@@ -97,6 +97,7 @@ def predict(text, model_id, tokenizer_id, label_names):
 def predict_illframes(text, language, domain):
     #DEBUG
     cache_dir = os.path.expanduser("~/.cache/huggingface")
+    shutil.rmtree("./data/")
     if os.path.exists(cache_dir):
         shutil.rmtree(cache_dir)
     #END DEBUG