Akseluhr commited on
Commit
a07082e
1 Parent(s): 5a22b5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,6 +4,7 @@ import gradio as gr
4
  pipe = pipeline(model="Akseluhr/whisper-small-sv-SE-auhr-v2")
5
 
6
  def get_audio(url):
 
7
  yt = YouTube(url)
8
  video = yt.streams.filter(only_audio=True).first()
9
  out_file=video.download(output_path=".")
@@ -17,13 +18,14 @@ def get_text(url):
17
  result = model.transcribe(get_audio(url))
18
  return result['text']
19
 
20
- def transcribe(rec=None, file=None, URL=None):
21
  if rec is not None:
22
  audio = rec
23
  elif file is not None:
24
  audio = file
25
  elif url is not None:
26
- audio = get_audio(URL)
 
27
  else:
28
  return "Provide a recording or a file."
29
 
 
4
  pipe = pipeline(model="Akseluhr/whisper-small-sv-SE-auhr-v2")
5
 
6
  def get_audio(url):
7
+ print(url)
8
  yt = YouTube(url)
9
  video = yt.streams.filter(only_audio=True).first()
10
  out_file=video.download(output_path=".")
 
18
  result = model.transcribe(get_audio(url))
19
  return result['text']
20
 
21
+ def transcribe(rec=None, file=None, url=None):
22
  if rec is not None:
23
  audio = rec
24
  elif file is not None:
25
  audio = file
26
  elif url is not None:
27
+ print(url)
28
+ audio = get_audio(url)
29
  else:
30
  return "Provide a recording or a file."
31