Upload app.py with huggingface_hub
Browse files
app.py
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
from transformers import pipeline
|
| 3 |
+
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
|
| 4 |
+
def summarize_text(text, max_length, min_length):
|
| 5 |
+
summary = summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)
|
| 6 |
+
return summary[0]['summary_text']
|
| 7 |
+
demo = gr.Interface(
|
| 8 |
+
fn=summarize_text,
|
| 9 |
+
inputs=[
|
| 10 |
+
gr.Textbox(
|
| 11 |
+
label="요약할 텍스트를 입력하세요",
|
| 12 |
+
placeholder="긴 텍스트를 입력하세요...",
|
| 13 |
+
lines=10
|
| 14 |
+
),
|
| 15 |
+
gr.Slider(
|
| 16 |
+
minimum=50,
|
| 17 |
+
maximum=500,
|
| 18 |
+
value=130,
|
| 19 |
+
step=10,
|
| 20 |
+
label="최대 길이"
|
| 21 |
+
),
|
| 22 |
+
gr.Slider(
|
| 23 |
+
minimum=10,
|
| 24 |
+
maximum=200,
|
| 25 |
+
value=30,
|
| 26 |
+
step=10,
|
| 27 |
+
label="최소 길이"
|
| 28 |
+
)
|
| 29 |
+
],
|
| 30 |
+
outputs=gr.Textbox(label="요약 결과"),
|
| 31 |
+
title="텍스트 요약기",
|
| 32 |
+
theme="soft",
|
| 33 |
+
examples=[
|
| 34 |
+
["""인공지능(AI)은 인간의 학습능력, 추론능력, 지각능력, 언어이해능력 등을 컴퓨터 프로그램으로 실현한 기술이다.
|
| 35 |
+
인공지능은 컴퓨터가 인간의 지능적인 행동을 모방할 수 있도록 하는 것을 목표로 한다.
|
| 36 |
+
이는 학습, 문제해결, 패턴인식, 음성인식, 시각적 지각, 의사결정 등을 포함한다.
|
| 37 |
+
최근에는 딥러닝과 기계학습의 발전으로 인공지능의 성능이 크게 향상되었으며,
|
| 38 |
+
다양한 분야에서 활용되고 있다.""", 100, 30],
|
| 39 |
+
]
|
| 40 |
+
)
|
| 41 |
+
|
| 42 |
+
if __name__ == '__main__':
|
| 43 |
+
demo.launch()
|