File size: 469 Bytes
5fe3014
 
 
7b68224
 
5fe3014
 
 
 
 
 
 
17f4304
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from transformers import pipeline, Conversation
import gradio as gr

summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

def summarizer_bot(message, history):
    return summarizer(message, min_length=5, max_length=140)[0]['summary_text']

demo_summarizer = gr.ChatInterface(
    summarizer_bot, title="Summarizer Chatbot",
    description="Enter your text, and the chatbot will return the summarized version.")

demo_summarizer.launch(share=True)