poltextlab commited on
Commit
9c6d143
·
verified ·
1 Parent(s): dbf346a

Update interfaces/illframes.py

Browse files
Files changed (1) hide show
  1. interfaces/illframes.py +1 -9
interfaces/illframes.py CHANGED
@@ -94,15 +94,7 @@ def predict(text, model_id, tokenizer_id, label_names):
94
  output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
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
- shutil.rmtree("/data/")
101
- if os.path.exists(cache_dir):
102
- shutil.rmtree(cache_dir)
103
- #END DEBUG
104
-
105
-
106
  domain = domains[domain]
107
  model_id = build_huggingface_path(domain)
108
  tokenizer_id = "xlm-roberta-large"
 
94
  output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id}">{model_id}</a> model.</p>'
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"