summerizer / app.py
akumar552's picture
Update app.py
0df5f7f
raw
history blame contribute delete
No virus
595 Bytes
from transformers import BertTokenizer
import os
import io
import torch
from transformers import pipeline
from transformers import AutoModel
acc_token='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'
get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
def summarize(input):
output = get_completion(input)
return output[0]['summary_text']
import gradio as gr
def summarize(input):
output = get_completion(input)
return output[0]['summary_text']
gr.close_all()
demo = gr.Interface(fn=summarize, inputs="text", outputs="text")
demo.launch(share=True)