mirari commited on
Commit
f0e5a8c
·
verified ·
1 Parent(s): 111d03e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,14 +1,14 @@
1
  import gradio as gr
2
  import whisper
3
- #from pytube import YouTube
4
- from pytubefix import YouTube
5
- from pytubefix.cli import on_progress
6
 
7
  loaded_model = whisper.load_model("medium")
8
  current_size = 'medium'
9
  def inference(link):
10
- #yt = YouTube(link)
11
- yt = YouTube(link, on_progress_callback=on_progress, use_po_token=True)
12
  global audio_stream
13
  audio_stream = yt.streams.filter(only_audio=True, file_extension='mp4').first()
14
  path = audio_stream.download()
 
1
  import gradio as gr
2
  import whisper
3
+ from pytube import YouTube
4
+ #from pytubefix import YouTube
5
+ #from pytubefix.cli import on_progress
6
 
7
  loaded_model = whisper.load_model("medium")
8
  current_size = 'medium'
9
  def inference(link):
10
+ yt = YouTube(link)
11
+ #yt = YouTube(link, on_progress_callback=on_progress, use_po_token=True)
12
  global audio_stream
13
  audio_stream = yt.streams.filter(only_audio=True, file_extension='mp4').first()
14
  path = audio_stream.download()