Spaces:
Sleeping
Sleeping
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()
|