Spaces:
Runtime error
Runtime error
File size: 1,900 Bytes
c776449 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
from youtube_transcript_api import YouTubeTranscriptApi
from youtube_transcript_api.formatters import TextFormatter
from transformers import pipeline
import torch
import gradio as gr
import sys
def get_video_id(url):
"""Extracts video ID from YouTube URL"""
if 'youtube.com' in url:
# Extracting video ID from URL
return url.split('v=')[1].split('&')[0]
elif 'youtu.be' in url:
# Extracting video ID from shortened URL
return url.split('/')[-1]
else:
raise ValueError("Invalid YouTube URL")
def get_summary(transcript):
model_name="Falconsai/text_summarization"
text_summary_pipeline = pipeline("summarization", model=model_name, device=-1, torch_dtype=torch.bfloat16)
# use device=-1 for CPU and device=0 for GPU (using litserve we could automate it, but without using a wrapper class 'device' has to be manually specified)
video_summary = text_summary_pipeline(transcript)
return video_summary[0]["summary_text"]
def get_transcript_summary(url):
"""Fetches and prints the transcript of a YouTube video given its URL"""
try:
video_id = get_video_id(url)
transcript = YouTubeTranscriptApi.get_transcript(video_id)
formatter = TextFormatter()
formatted_transcript = formatter.format_transcript(transcript)
transcript_summary = get_summary(formatted_transcript)
return transcript_summary
except Exception as e:
print("An error occurred:", e)
gr.close_all()
# version 0.1
# demo = gr.Interface(fn=summary, inputs="text", outputs="text")
# version 0.2
demo = gr.Interface(
fn=get_transcript_summary,
inputs=[gr.Textbox(label="Youtube Video URL", lines=2)],
outputs=[gr.Textbox(label="YouTube Video's Summary", lines=6)],
title="GenAI YouTube Video Summarizer",
description="This App Summarizes Youtube Videos")
demo.launch() |