specify folder

interfaces/illframes.py

@@ -58,7 +58,7 @@ def predict(text, model_id, tokenizer_id, label_names):
 
     # --- DEBUG ---
     
-    disk_space = get_disk_space()
+    disk_space = get_disk_space('/data/')
     print("Disk Space Info:")
     for key, value in disk_space.items():
         print(f"{key}: {value}")
@@ -68,7 +68,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)
     except:
-        disk_space = get_disk_space()
+        disk_space = get_disk_space('/data/')
         print("Disk Space Error:")
         for key, value in disk_space.items():
             print(f"{key}: {value}")