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

remove models

Browse files
Files changed (1) hide show
  1. 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
- #os.system('rm -r /data/')
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