File size: 683 Bytes
557d7e6
0ed5942
557d7e6
 
0ed5942
557d7e6
 
 
0ed5942
557d7e6
0ed5942
557d7e6
0ed5942
 
 
 
 
 
 
 
 
 
557d7e6
0ed5942
557d7e6
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
import os

# Install dependencies manually if not found
os.system("pip install torch torchvision torchaudio transformers gradio")

import torch
import gradio as gr
from transformers import pipeline

text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)

def summary(input):
    output = text_summary(input)
    return output[0]['summary_text']

demo = gr.Interface(
    fn=summary,
    inputs=[gr.Textbox(label="Input text to summarize", lines=7)],
    outputs=[gr.Textbox(label="Summarized text", lines=4)],
    title="@cygon: Text summarizer",
    description="SIMPLE UNPAID APPLICATION FOR TEXT SUMMARIZING"
)

demo.launch()