Spaces:
Build error
Build error
| #!pip install -q transformers | |
| #!pip install -q youtube_transcript_api | |
| import gradio as gr | |
| from transformers import pipeline | |
| from youtube_transcript_api import YouTubeTranscriptApi | |
| def TextScript_demo(input_text): | |
| #youtube_video = "https://www.youtube.com/watch?v=7gTpHugdUWc" | |
| youtube_video=input_text | |
| video_id = youtube_video.split("=")[1] | |
| YouTubeTranscriptApi.get_transcript(video_id) | |
| transcript = YouTubeTranscriptApi.get_transcript(video_id) | |
| result = "" | |
| for i in transcript: | |
| result += ' ' + i['text'] | |
| #print(result) | |
| # print(len(result)) | |
| summarizer = pipeline('summarization') | |
| num_iters = int(len(result)/1000) | |
| summarized_text = [] | |
| for i in range(0, num_iters + 1): | |
| start = 0 | |
| start = i * 1000 | |
| end = (i + 1) * 1000 | |
| #print("input text \n" + result[start:end]) | |
| out = summarizer(result[start:end]) | |
| #out = out[0] | |
| #out = out['summary_text'] | |
| #print("Summarized text\n"+out) | |
| #summarized_text.append(out) | |
| return out | |
| title ='Text Script Generation from YouTube Link' | |
| description = "Extract Script" | |
| textScript_demo= gr.Interface(fn=TextScript_demo, inputs = 'text', outputs='text', title = title, description = description) | |