themanas021's picture
Create app.py
2260d33
raw
history blame
899 Bytes
import gradio as gr
from summarize import Summarizer
interface = gr.Interface(fn = Summarizer,
inputs = [gr.inputs.Textbox(lines=2,
placeholder="Enter your link...",
label='YouTube Video Link'),
gr.inputs.Radio(["mT5", "BART", "Pegasus"], type="value", label='Model')],
outputs = [gr.outputs.Textbox(
label="Summary")],
title = "Youtube Summarizer",
examples = [['https://www.youtube.com/watch?v=A4OmtyaBHFE', 'mT5'],
['https://www.youtube.com/watch?v=cU6xVZfkcgo', 'mT5']],
enable_queue=True)
interface.launch(debug=True)