File size: 469 Bytes
3758e24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from transformers import pipeline
import gradio as gr

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

# Function for summarization
def summarize_text(text):
    return summarizer(text, max_length=60, min_length=20, do_sample=False)[0]["summary_text"]

# Gradio interface
iface = gr.Interface(
    fn=summarize_text,
    inputs="text",
    outputs="text",
    title="Summarizer LLM"
)

# Launch app
iface.launch()