poltextlab commited on
Commit
a8ac4e6
·
verified ·
1 Parent(s): a5419e9

add disk space check

Browse files
Files changed (1) hide show
  1. interfaces/emotion9.py +7 -0
interfaces/emotion9.py CHANGED
@@ -9,6 +9,8 @@ from huggingface_hub import HfApi
9
 
10
  from label_dicts import EMOTION9_LABEL_NAMES
11
 
 
 
12
  HF_TOKEN = os.environ["hf_read"]
13
 
14
  languages = [
@@ -47,6 +49,11 @@ def predict(text, model_id, tokenizer_id):
47
  def predict_e6(text, language, domain):
48
  model_id = build_huggingface_path(language)
49
  tokenizer_id = "xlm-roberta-large"
 
 
 
 
 
50
  return predict(text, model_id, tokenizer_id)
51
 
52
  demo = gr.Interface(
 
9
 
10
  from label_dicts import EMOTION9_LABEL_NAMES
11
 
12
+ from utils import is_dick_space
13
+
14
  HF_TOKEN = os.environ["hf_read"]
15
 
16
  languages = [
 
49
  def predict_e6(text, language, domain):
50
  model_id = build_huggingface_path(language)
51
  tokenizer_id = "xlm-roberta-large"
52
+
53
+ if is_disk_full():
54
+ os.system('rm -rf /data/models*')
55
+ os.system('rm -r ~/.cache/huggingface/hub')
56
+
57
  return predict(text, model_id, tokenizer_id)
58
 
59
  demo = gr.Interface(