KAHRAMAN42 commited on
Commit
4d6f55d
·
1 Parent(s): 8a2f010

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
  from youtube_transcript_api import YouTubeTranscriptApi
4
 
 
5
  def youtube(url):
6
  video_id=url.split("=")[1]
7
  transcript=YouTubeTranscriptApi.get_transcript(video_id)
@@ -10,7 +11,10 @@ def youtube(url):
10
  result+=i["text"]+" "+"\n"
11
  summarization=pipeline("summarization",model="facebook/bart-large-cnn",max_length=60,min_length=10)
12
  list_summarize=summarization(result)[0]["summary_text"].split(". ")
13
- return list_summarize[0]
 
 
 
14
 
15
  iface = gr.Interface(
16
  fn=youtube,
 
2
  from transformers import pipeline
3
  from youtube_transcript_api import YouTubeTranscriptApi
4
 
5
+
6
  def youtube(url):
7
  video_id=url.split("=")[1]
8
  transcript=YouTubeTranscriptApi.get_transcript(video_id)
 
11
  result+=i["text"]+" "+"\n"
12
  summarization=pipeline("summarization",model="facebook/bart-large-cnn",max_length=60,min_length=10)
13
  list_summarize=summarization(result)[0]["summary_text"].split(". ")
14
+ result=""
15
+ for m in list_summarize:
16
+ result+=m+".\n"
17
+ return result
18
 
19
  iface = gr.Interface(
20
  fn=youtube,