Spaces:
Running
Running
clear cache
Browse files- interfaces/illframes.py +7 -0
interfaces/illframes.py
CHANGED
@@ -95,6 +95,13 @@ def predict(text, model_id, tokenizer_id, label_names):
|
|
95 |
return output_pred, output_info
|
96 |
|
97 |
def predict_illframes(text, language, domain):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
98 |
domain = domains[domain]
|
99 |
model_id = build_huggingface_path(domain)
|
100 |
tokenizer_id = "xlm-roberta-large"
|
|
|
95 |
return output_pred, output_info
|
96 |
|
97 |
def predict_illframes(text, language, domain):
|
98 |
+
#DEBUG
|
99 |
+
cache_dir = os.path.expanduser("~/.cache/huggingface")
|
100 |
+
if os.path.exists(cache_dir):
|
101 |
+
shutil.rmtree(cache_dir)
|
102 |
+
#END DEBUG
|
103 |
+
|
104 |
+
|
105 |
domain = domains[domain]
|
106 |
model_id = build_huggingface_path(domain)
|
107 |
tokenizer_id = "xlm-roberta-large"
|