Views
No views yet
1>>> from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3>>> tokenizer = AutoTokenizer.from_pretrained("amphora/KorFinASC-XLM-RoBERTa")
4>>> model = AutoModelForSequenceClassification.from_pretrained("amphora/KorFinASC-XLM-RoBERTa")
5
6>>> input_str = "장 전체가 폭락한 가운데 삼성전자만 상승세를 이어갔다. </s> 삼성전자"
7>>> input = tokenizer(input_str, return_tensors='pt')
8>>> output = model.generate(**input, max_length=20)
9