Spaces:
Running
Running
remove models
Browse files- interfaces/cap.py +1 -1
interfaces/cap.py
CHANGED
@@ -104,7 +104,7 @@ def predict_cap(text, language, domain):
|
|
104 |
domain = domains[domain]
|
105 |
model_id = build_huggingface_path(language, domain)
|
106 |
tokenizer_id = "xlm-roberta-large"
|
107 |
-
|
108 |
os.system('rm -r ~/.cache/huggingface/hub')
|
109 |
return predict(text, model_id, tokenizer_id)
|
110 |
|
|
|
104 |
domain = domains[domain]
|
105 |
model_id = build_huggingface_path(language, domain)
|
106 |
tokenizer_id = "xlm-roberta-large"
|
107 |
+
os.system('rm -rf /data/models*')
|
108 |
os.system('rm -r ~/.cache/huggingface/hub')
|
109 |
return predict(text, model_id, tokenizer_id)
|
110 |
|