Spaces:
Running
Running
add disk space check
Browse files- 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(
|