Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
2
3model = AutoModelForCausalLM.from_pretrained("Ja-ck/mistral-7b-tag-extractor-v1")
4tokenizer = AutoTokenizer.from_pretrained("Ja-ck/mistral-7b-tag-extractor-v1")
5
6gen_config = GenerationConfig(
7 pad_token_id=tokenizer.pad_token_id,
8 temperature=0.2,
9 top_p=1,
10 top_k=40,
11 num_beams=1,
12 repetition_penalty=1.11,
13 do_sample=True,
14 eos_token_id=tokenizer.eos_token_id,
15)
16
17messages = [
18 {"role": "user", "content": """“이 책을 읽기 전으로 되돌아갈 수 없다!”
19공감을 불러일으키는 걸작인가, 피하고 싶은 문제작인가?
20
21누적 판매 50만 부 돌파, 화제의 베스트셀러!
222023년 영화 〈정욕〉 일본 개봉
23제34회 시바타 렌자부로상 수상
242022년 서점 대상 4위
25오디오 북 대상 2023 무제한 청취 부문 대상
26〈다빈치〉 플래티넘 도서 OF THE YEAR 2021
27〈다빈치〉 BOOK OF THE YEAR 2023 문고 1위
28일본 서평 사이트 북로그 2021년 연간 등록 1위, ‘#최고의책’ 최다 등록
29일본 최대 서점 기노쿠니야 선정 2022년 베스트셀러 2위
30
312021년 출간 이후, 일본 최고 문제작이자 화제작으로 떠오른 장편소설, ‘《정욕正欲》’이 리드비에서 소개된다. 최연소 남성 나오키상 수상 작가 아사이 료의 데뷔 10주년 기념작이기도 한 이 작품은 성적 욕망을 뜻하는 ‘정욕(情慾)’, 마음속의 욕구를 다룬 ‘정욕(情欲)’이 아닌 ‘바른 욕망’이란 뜻의 ‘正欲’이란 한자를 제목으로 삼고 있다.
32
33《정욕》은 ‘다양성 존중의 시대’를 살고 있다고 자부하는 우리가 받아들일 수 있는 범위는 과연 어디까지인지, 과감하고도 도발적인 질문을 던진다. ‘다양성’에 대한 일반인의 상식을 뒤엎는 파격적인 전개로 격렬한 찬반 논쟁을 불러일으킨 《정욕》은 제34회 시바타 렌자부로상, 2022년 서점 대상 4위 등 비평적 찬사는 물론, 2021년부터 현재까지 각종 도서 랭킹 상위에 오르며 일본 문학계 최고의 화제작으로 자리 잡았다.
34
35《정욕》은 2023년 이나가키 고로, 아라가키 유이 주연 영화로 제작됐으며, 영화 또한 소설 못지않은 화제를 불러일으키며 제36회 도쿄 국제 영화제에서 최우수 감독상, 관객상을 수상했다. 영화 〈정욕〉은 2024년, 국내 개봉을 앞두고 있다."""}
36]
37
38inputs = tokenizer(
39[
40 tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True),
41], return_tensors='pt').to("cuda")
42
43outputs = model.generate(**inputs, generation_config=gen_config, max_new_tokens=512,use_cache=True)
44result = tokenizer.batch_decode(outputs,skip_special_tokens=True)
45result1['### 질문: 아래 텍스트에서 주제어를 최대 20개 추출해주세요. 출력 형식은 {{"topics": [ "주제어1","주제어2", "주제어3", ...] }}과 동일한 형식으로 합니다.
2짧은 단어 보다는 두개 이상의 단어들로 주제어를 추출합니다. 명사나 명사구가 좋습니다. 중복되는 주제어는 제외합니다.
3텍스트: “이 책을 읽기 전으로 되돌아갈 수 없다!”\n공감을 불러일으키는 걸작인가, 피하고 싶은 문제작인가?\n\n누적 판매 50만 부 돌파, 화제의 베스트셀러!\n2023년 영화 〈정욕〉 일본 개봉\n제34회 시바타 렌자부로상 수상\n2022년 서점 대상 4위\n오디오 북 대상 2023 무제한 청취 부문 대상\n〈다빈치〉 플래티넘 도서 OF THE YEAR 2021\n〈다빈치〉 BOOK OF THE YEAR 2023 문고 1위\n일본 서평 사이트 북로그 2021년 연간 등록 1위, ‘#최고의책’ 최다 등록\n일본 최대 서점 기노쿠니야 선정 2022년 베스트셀러 2위\n\n2021년 출간 이후, 일본 최고 문제작이자 화제작으로 떠오른 장편소설, ‘《정욕正欲》’이 리드비에서 소개된다. 최연소 남성 나오키상 수상 작가 아사이 료의 데뷔 10주년 기념작이기도 한 이 작품은 성적 욕망을 뜻하는 ‘정욕(情慾)’, 마음속의 욕구를 다룬 ‘정욕(情欲)’이 아닌 ‘바른 욕망’이란 뜻의 ‘正欲’이란 한자를 제목으로 삼고 있다.\n\n《정욕》은 ‘다양성 존중의 시대’를 살고 있다고 자부하는 우리가 받아들일 수 있는 범위는 과연 어디까지인지, 과감하고도 도발적인 질문을 던진다. ‘다양성’에 대한 일반인의 상식을 뒤엎는 파격적인 전개로 격렬한 찬반 논쟁을 불러일으킨 《정욕》은 제34회 시바타 렌자부로상, 2022년 서점 대상 4위 등 비평적 찬사는 물론, 2021년부터 현재까지 각종 도서 랭킹 상위에 오르며 일본 문학계 최고의 화제작으로 자리 잡았다.\n\n《정욕》은 2023년 이나가키 고로, 아라가키 유이 주연 영화로 제작됐으며, 영화 또한 소설 못지않은 화제를 불러일으키며 제36회 도쿄 국제 영화제에서 최우수 감독상, 관객상을 수상했다. 영화 〈정욕〉은 2024년, 국내 개봉을 앞두고 있다.
4
5### 답변: {"topics": ["문제작", "화제의 베스트셀러", "성적 욕망", "다양성 존중", "영화 애호가"]}']