File size: 1,232 Bytes
847634c
 
214a767
f1a59cf
 
 
 
847634c
f1a59cf
 
 
 
214a767
 
f1a59cf
214a767
 
847634c
 
 
 
 
e2f2df1
01d910c
847634c
f1a59cf
847634c
 
 
 
 
 
 
 
01d910c
847634c
 
 
 
214a767
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import gradio as gr
from gradio.mix import Parallel, Series

from transformers import pipeline

summarizer = pipeline("summarization", model="VietAI/vit5-large-vietnews-summarization")

 
def summarize(inp):
    text = "vietnews: " + inp + " </s>"
    res = summarizer(
        text,
        max_length=256,
        early_stopping=True,
    )[0]['summary_text']
    return res



sample_url = [['VietAI là tổ chức phi lợi nhuận với sứ mệnh ươm mầm tài năng về trí tuệ nhân tạo và xây dựng một cộng đồng các chuyên gia trong lĩnh vực trí tuệ nhân tạo đẳng cấp quốc tế tại Việt Nam.'],
]

article = "<p style='text-align: center'><a href='https://vietai.org' target='_blank'>by VietAI Research</a> | <a href='https://github.com/vietai/ViT5' target='_blank'>Github</a> | Contact: <a href='mailto:heraclex12@gmail.com' target='_blank'>Hieu Tran</a></p></center></p>"


iface = gr.Interface(fn=summarize, 
  inputs = gr.inputs.Textbox(
      lines = 5,
      label = 'Enter an article...'
  ),
  outputs = 'text',
  title = 'Vi(etnamese)T5 Abstractive Summarization',
  theme = 'grass',
  layout = 'horizontal',
  article=article,
  examples=sample_url)

iface.launch()