Javed7 commited on
Commit
3219217
·
1 Parent(s): b195dad

removed summarizer

Browse files
Files changed (1) hide show
  1. app.py +11 -27
app.py CHANGED
@@ -1,46 +1,30 @@
1
  import whisper
2
  from pytube import YouTube
3
- from transformers import pipeline
4
  import gradio as gr
5
  import os
6
 
7
  model = whisper.load_model("large")
8
- summarizer = pipeline("summarization")
9
 
10
  def get_audio(url):
11
  yt = YouTube(url)
12
  video = yt.streams.filter(only_audio=True).first()
13
- out_file=video.download(output_path=".")
14
  base, ext = os.path.splitext(out_file)
15
- new_file = base+'.mp3'
16
  os.rename(out_file, new_file)
17
- a = new_file
18
- return a
19
 
20
  def get_text(url):
21
  result = model.transcribe(get_audio(url))
22
  return result['text']
23
 
24
- def get_summary(url):
25
- article = get_text(url)
26
- b = summarizer(article)
27
- b = b[0]['summary_text']
28
- return b
29
-
30
  with gr.Blocks() as demo:
31
  gr.Markdown("<h1><center>Youtube video transcription with OpenAI's Whisper</center></h1>")
32
- gr.Markdown("<center>Enter the link of any youtube video to get the transcription of the video and a summary of the video in the form of text.</center>")
33
- with gr.Tab('Get the transcription of any Youtube video'):
34
- with gr.Row():
35
- input_text_1 = gr.Textbox(placeholder='Enter the Youtube video URL', label='URL')
36
- output_text_1 = gr.Textbox(placeholder='Transcription of the video', label='Transcription')
37
- result_button_1 = gr.Button('Get Transcription')
38
- with gr.Tab('Summary of Youtube video'):
39
- with gr.Row():
40
- input_text = gr.Textbox(placeholder='Enter the Youtube video URL', label='URL')
41
- output_text = gr.Textbox(placeholder='Summary text of the Youtube Video', label='Summary')
42
- result_button = gr.Button('Get Summary')
43
-
44
- result_button.click(get_summary, inputs = input_text, outputs = output_text)
45
- result_button_1.click(get_text, inputs = input_text_1, outputs = output_text_1)
46
- demo.launch(debug=True)
 
1
  import whisper
2
  from pytube import YouTube
 
3
  import gradio as gr
4
  import os
5
 
6
  model = whisper.load_model("large")
 
7
 
8
  def get_audio(url):
9
  yt = YouTube(url)
10
  video = yt.streams.filter(only_audio=True).first()
11
+ out_file = video.download(output_path=".")
12
  base, ext = os.path.splitext(out_file)
13
+ new_file = base + '.mp3'
14
  os.rename(out_file, new_file)
15
+ return new_file
 
16
 
17
  def get_text(url):
18
  result = model.transcribe(get_audio(url))
19
  return result['text']
20
 
 
 
 
 
 
 
21
  with gr.Blocks() as demo:
22
  gr.Markdown("<h1><center>Youtube video transcription with OpenAI's Whisper</center></h1>")
23
+ gr.Markdown("<center>Enter the link of any youtube video to get the transcription of the video in the form of text.</center>")
24
+ with gr.Row():
25
+ input_text_1 = gr.Textbox(placeholder='Enter the Youtube video URL', label='URL')
26
+ output_text_1 = gr.Textbox(placeholder='Transcription of the video', label='Transcription')
27
+ result_button_1 = gr.Button('Get Transcription')
28
+
29
+ result_button_1.click(get_text, inputs=input_text_1, outputs=output_text_1)
30
+ demo.launch(debug=True)