Spaces:
Runtime error
Runtime error
removed summarizer
Browse files
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 |
-
|
| 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
|
| 33 |
-
with gr.
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|