File size: 1,241 Bytes
0e5d440
 
 
 
 
 
 
 
 
 
 
94c979f
b59c9a5
3b7132c
 
 
 
 
b59c9a5
0e5d440
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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"], type="value", label='Model')],
                        outputs = [gr.outputs.Textbox(
                                                      label="Summary")],
                         
                        title = "Video Summary Generator",
                        examples = [
                                   ['https://www.youtube.com/watch?v=eEE2w4Pe2uY', 'BART'],
                                   ['https://www.youtube.com/watch?v=dmdatn8r3cU', 'BART'],
                                   ['https://www.youtube.com/watch?v=-TNr_UvpcYU', 'BART'],
                                   ['https://www.youtube.com/shorts/DY2Eo25XDQI', 'BART'],
                                   ['https://www.youtube.com/shorts/FCBO0mzfP7o', 'mT5'],
                                   ],
                        enable_queue=True)

interface.launch(debug=True)