Использование локально сохраненной модели SetFit
Я сохранил модель SetFit локально, теперь хочу использовать ее для прогнозирования (для развертывания модели). Я попробовал следующий код:
from setfit import SetFitModel
model = SetFitModel.from_pretrained("CyberBullying/cyberbullying-classification-setfit-model-v1")
preds = model(["i loved the spiderman movie!", "Another Idiot ...is India really Full of them. It's has nothing to do with BJP / CONgress or Hindu-Muslim, it's about Rich/Poor. You all are so uneducated , but literate ."])
print('========================Predictions========================')
print(preds)
но появляется следующая ошибка:
Обратная трассировка (последний вызов): файл «F:\Chahinez RSI\prediction.py», строка 3, в model = SetFitModel.from_pretrained(»CyberBullying/cyberbullying-classification-setfit-model-v1») Файл «C:\Users\DELL\AppData\Local\Programs\Python\Python310\lib\site-packages\huggingface_hub\utils_validators.py", строка 118, в _inner_fn верните fn(*args, **kwargs) Файл "C:\Users\DELL\AppData\Local\Programs\Python\Python310\lib\site-packages\huggingface_hub\hub_mixin.py", строка 159, в from_pretrained
return cls._from_pretrained( File "C:\Users\DELL\AppData\Local\Programs\Python \Python310\lib\site-packages\setfit\modeling.py", строка 549, в файле _from_pretrainedmodel_head = joblib.load(model_head_file) "C:\Users\DELL\AppData\Local\Programs\Python\Python310\lib\site-packages\joblib\numpy_pickle.py", строка 587, при загрузке obj = _unpickle(fobj, filename, mmap_mode) Файл "C:\Users\DELL\AppData\Local\Programs\Python\Python310\lib\site-packages\joblib\numpy_pickle.py", строка 506, в _unpickle obj = unpickler.load() Файл "C:\Users\DELL\AppData\Local\Programs\Python\Python310\lib\pickle.py", строка 1213, в загрузить диспетчерский ключ [0]KeyError: 0