department_classifier_space / predict_dept_model.py

Commit History

set force_download=True for tokenizer and the model
76ceff2
verified

mr-kush commited on

fix the self.cache_dir not found build error
1509190
verified

mr-kush commited on

update the path and ensure the cache folder exists
6ec367f
verified

mr-kush commited on

change the load model cache dir to /tmp/huggingface
985da36
verified

mr-kush commited on

replace repalce return_all_tensors=True with top_k = None
e787a7e
verified

mr-kush commited on

Add DepartmentPredictor class for department classification model
b3cf3ad

mr-kush commited on