Views
No views yet
1from transformers import PreTrainedTokenizerFast, BartForConditionalGeneration
2
3# Load Model and Tokenizer
4tokenizer = PreTrainedTokenizerFast.from_pretrained("EbanLee/kobart-summary-v2")
5model = BartForConditionalGeneration.from_pretrained("EbanLee/kobart-summary-v2")
6
7# Encoding
8input_text = "일반적으로 로컬푸드 농산물은 도매시장 경매가보다 높고 소매가보다 낮은 수준에서 가격이 결정된다. 농가들이 자율적으로 가격을 결정하고 있지만, 시장가격을 참고하기 때문에 대체로 적정한 가격이 설정된다. 문제는 태풍, 폭우, 폭염 또는 공급과잉에 따른 시장가격 등락이 심해질 때이다. 완주군 로컬푸드 협동조합은 위와 같이 외부요인에 의해 농산물 가격에 급등락이 발생했을 때에도 연중 일정 수준의 가격으로 판매되도록 유통 안정 기금을 운용하고 있다. 그리고 농산물 출하자에게 사전에 동의를 구해 일부 관리 품목은 가격 상하한 및 판매량을 조절해 납품하도록 교육하고 있다."
9input_ids = tokenizer.encode(input_text, return_tensors="pt", padding="max_length", truncation=True, max_length=1026)
10
11# Generate Summary Text Ids
12summary_text_ids = model.generate(
13input_ids=input_ids,
14bos_token_id=model.config.bos_token_id,
15eos_token_id=model.config.eos_token_id,
16length_penalty=1.5,
17max_length=256,
18min_length=12,
19num_beams=6,
20repetition_penalty=1.5,
21)
22
23# Decoding Text Ids
24print(tokenizer.decode(summary_text_ids[0], skip_special_tokens=True))
25