archit11 commited on
Commit
6d1feb7
·
verified ·
1 Parent(s): 503b72f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import os
2
-
3
  import uuid
4
  import yt_dlp as youtube_dl
5
  from typing import Generator
@@ -63,6 +63,7 @@ class YouTubeTranscriber:
63
  return df
64
 
65
  # Function to be called by the Gradio interface
 
66
  def transcribe_youtube_video(url: str, model_path: str = "distil-large-v2") -> str:
67
  yt_transcriber = YouTubeTranscriber(model_path)
68
  audio_path = yt_transcriber.download_audio(url)
 
1
  import os
2
+ import spaces
3
  import uuid
4
  import yt_dlp as youtube_dl
5
  from typing import Generator
 
63
  return df
64
 
65
  # Function to be called by the Gradio interface
66
+ @spaces.GPU
67
  def transcribe_youtube_video(url: str, model_path: str = "distil-large-v2") -> str:
68
  yt_transcriber = YouTubeTranscriber(model_path)
69
  audio_path = yt_transcriber.download_audio(url)