papasega commited on
Commit
5255e11
·
verified ·
1 Parent(s): e17dd58

Update app_app_youtube_trancribe.py

Browse files
Files changed (1) hide show
  1. app_app_youtube_trancribe.py +6 -2
app_app_youtube_trancribe.py CHANGED
@@ -3,8 +3,12 @@ from pytube import YouTube
3
  from transformers import pipeline
4
  import gradio as gr
5
  import os
 
 
 
 
6
 
7
- model = whisper.load_model("base")
8
  summarizer = pipeline("summarization")
9
 
10
  def get_audio(url):
@@ -28,7 +32,7 @@ def get_summary(url):
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 and a summary of the video in the form of text.</center>")
33
  with gr.Tab('Get the transcription of any Youtube video'):
34
  with gr.Row():
 
3
  from transformers import pipeline
4
  import gradio as gr
5
  import os
6
+ from speechbrain.inference.ASR import EncoderASR
7
+ import gradio as gr
8
+
9
+ model = EncoderASR.from_hparams("speechbrain/asr-wav2vec2-dvoice-wolof")
10
 
11
+ #model = whisper.load_model("base")
12
  summarizer = pipeline("summarization")
13
 
14
  def get_audio(url):
 
32
  return b
33
 
34
  with gr.Blocks() as demo:
35
+ gr.Markdown("<h1><center>Youtube video transcription with in wolof by Papa Sega</center></h1>")
36
  gr.Markdown("<center>Enter the link of any youtube video to get the transcription of the video and a summary of the video in the form of text.</center>")
37
  with gr.Tab('Get the transcription of any Youtube video'):
38
  with gr.Row():