Views
No views yet
1from transformers import pipeline
2
3# 모델 및 토크나이저 로드
4pipe_finetuned = pipeline("text-generation", model="your-username/gemma-2b-it-sum-ko-legal", tokenizer="your-username/gemma-2b-it-sum-ko-legal", max_new_tokens=512)
5
6# 요약할 텍스트 입력
7paragraph = """
8 한국의 법률안 검토 보고서 내용은 매우 복잡하고 긴 경우가 많습니다.
9 이러한 문서를 요약하여 주요 정보를 빠르게 파악하는 것이 중요합니다.
10"""
11
12# 요약 요청
13summary = pipe_finetuned(paragraph, do_sample=True, temperature=0.2, top_k=50, top_p=0.95)
14print(summary[0]["generated_text"])