File size: 545 Bytes
da94187
7785600
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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('&quot', '"')
  text = text.replace('&apos', "'")
  text = text.replace('&amp', '&')
  result = summarizer(text, min_length=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()