Views
No views yet
1from transformers import AutoTokenizer, AutoModel
2from app.ananke.model import XMLRoBERTaClassifier
3
4# 모델 로드
5model = XMLRoBERTaClassifier(model_dir="greensteel-frontend-material-classifier")
6
7# 예측
8result = model.predict("철ㄹ")
9print(result) # [{'label': '철', 'similarity': 100.0, 'match_type': 'exact_input'}]