File size: 658 Bytes
9214b38
91ba568
 
 
 
 
 
 
 
 
 
 
93c1088
91ba568
 
 
 
 
e91ce67
91ba568
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from transformers import pipeline
import gradio as gr

summarizer = pipeline('summarization', 
                      model='t5-base', 
                      tokenizer='t5-small', 
                      truncation=True, 
                      framework='tf')

def translate(text):
    text = text.replace('"', '"')
    text = text.replace(''', "'")
    text = text.replace('&', "&")
    result =summarizer(text, min_lenght=180, truncation=True)
    return result[0]['summary_text']

iface = gr.Interface(
    fn = translate,
    inputs = gr.inputs.Textbox(lines=10, placeholder='Enter text to summarize...'),
    outputs='Text'
)
iface.launch()