poltextlab commited on
Commit
dbb4136
·
verified ·
1 Parent(s): 7ac6eec

add is_disk_full func

Browse files
Files changed (1) hide show
  1. utils.py +11 -0
utils.py CHANGED
@@ -1,4 +1,6 @@
1
  import os
 
 
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
 
4
  from interfaces.cap import languages as languages_cap
@@ -30,3 +32,12 @@ def download_hf_models():
30
  for tokenizer_id in tokenizers:
31
  AutoTokenizer.from_pretrained(tokenizer_id)
32
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
+ import shutil
3
+
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
 
6
  from interfaces.cap import languages as languages_cap
 
32
  for tokenizer_id in tokenizers:
33
  AutoTokenizer.from_pretrained(tokenizer_id)
34
 
35
+
36
+ def is_disk_full(min_free_space_in_GB=10):
37
+ total, used, free = shutil.disk_usage("/")
38
+ free_gb = free / (1024 ** 3)
39
+
40
+ if free_gb >= min_free_space_in_GB:
41
+ return False
42
+ else:
43
+ return True