remove offloading

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", offload_folder="offload", token=HF_TOKEN)
+        model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", token=HF_TOKEN)
     except:
         disk_space = get_disk_space('/data/')
         print("Disk Space Error:")