File size: 515 Bytes
f163753
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

#packages
import gradio as gr
import torch
from transformers import pipeline

#%% Text Summarization
pipe = pipeline("summarization", model="Falconsai/text_summarization")

def summarize(text):
    result = pipe(text, min_length=5, max_length=20)
    return result[0]['summary_text']

title = "Text Summarization"
description = "This model summarizes an article (using  Falconsai/text_summarization model)"


gr.Interface(fn=summarize, inputs="text", outputs="text", title=title, description=description).launch()