nithinraok commited on
Commit
4ec34e8
·
verified ·
1 Parent(s): 52039c9

update_to_use_zeroGPU (#1)

Browse files

- update to use zeroGPU (dca363ae42b1d706fca4f63f4a2abf3b0b2ee17f)

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import shutil
2
  import gradio as gr
 
3
  import yt_dlp
4
  import os
5
  import tempfile
@@ -410,7 +411,7 @@ def download_youtube_audio(url, force_reload=False):
410
  except Exception as e:
411
  return None, f"❌ Download error: {str(e)}"
412
 
413
-
414
  def infer(audio_path, youtube_url, prompt_text):
415
  try:
416
  final_audio_path = None
 
1
  import shutil
2
  import gradio as gr
3
+ import spaces
4
  import yt_dlp
5
  import os
6
  import tempfile
 
411
  except Exception as e:
412
  return None, f"❌ Download error: {str(e)}"
413
 
414
+ @spaces.GPU
415
  def infer(audio_path, youtube_url, prompt_text):
416
  try:
417
  final_audio_path = None