mbCrypto commited on
Commit
198c7f7
Β·
1 Parent(s): 3e14b41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  # Initialize the space
2
  summarizeryt = pipeline("summarization", model="facebook/bart-large-cnn")
3
  summarizerbg = pipeline("summarization")
@@ -6,7 +10,6 @@ summarizertx = PegasusForConditionalGeneration.from_pretrained("google/pegasus-x
6
  # How to use: YTVideoToText("https://www.youtube.com/watch?v=jQL0ZeHtXFc")
7
  def YTVideoToText(video_link):
8
  # installing & importing libraries
9
- from transformers import pipeline
10
  from youtube_transcript_api import YouTubeTranscriptApi
11
 
12
  # fetching video transcript
@@ -39,7 +42,6 @@ def YTVideoToText(video_link):
39
  # How to use: postSummaryWithBart("https://ethereum.org/en/what-is-ethereum/")
40
  def postSummaryWithBart(blog_link):
41
  # importing libraries
42
- from transformers import pipeline
43
  from bs4 import BeautifulSoup
44
  import requests
45
 
@@ -86,7 +88,7 @@ def postSummaryWithBart(blog_link):
86
  # How to use: abstractiveSummaryWithPegasus("""Sample text to be summarized""")
87
  def abstractiveSummaryWithPegasus(words):
88
  # importing & loading model
89
- from transformers import PegasusForConditionalGeneration, PegasusTokenizer
90
  tokenizer = PegasusTokenizer.from_pretrained("google/pegasus-xsum")
91
 
92
  # perform summarization
@@ -108,9 +110,6 @@ def process(uri, mode):
108
  raise ValueError("Invalid mode")
109
 
110
 
111
- import gradio as gr
112
- from gradio.mix import Series
113
-
114
  with gr.Blocks() as ui:
115
  gr.Markdown("""
116
  ## Permet de faire le rΓ©sumΓ© d'une video youtube ou d'un article de blog
 
1
+ import gradio as gr
2
+ from transformers import pipeline, PegasusForConditionalGeneration
3
+
4
+
5
  # Initialize the space
6
  summarizeryt = pipeline("summarization", model="facebook/bart-large-cnn")
7
  summarizerbg = pipeline("summarization")
 
10
  # How to use: YTVideoToText("https://www.youtube.com/watch?v=jQL0ZeHtXFc")
11
  def YTVideoToText(video_link):
12
  # installing & importing libraries
 
13
  from youtube_transcript_api import YouTubeTranscriptApi
14
 
15
  # fetching video transcript
 
42
  # How to use: postSummaryWithBart("https://ethereum.org/en/what-is-ethereum/")
43
  def postSummaryWithBart(blog_link):
44
  # importing libraries
 
45
  from bs4 import BeautifulSoup
46
  import requests
47
 
 
88
  # How to use: abstractiveSummaryWithPegasus("""Sample text to be summarized""")
89
  def abstractiveSummaryWithPegasus(words):
90
  # importing & loading model
91
+ from transformers import PegasusTokenizer
92
  tokenizer = PegasusTokenizer.from_pretrained("google/pegasus-xsum")
93
 
94
  # perform summarization
 
110
  raise ValueError("Invalid mode")
111
 
112
 
 
 
 
113
  with gr.Blocks() as ui:
114
  gr.Markdown("""
115
  ## Permet de faire le rΓ©sumΓ© d'une video youtube ou d'un article de blog