Upload app.py

app.py

@@ -2,6 +2,7 @@ import os
 import torch
 import logging
 import yt_dlp
+import spaces
 import gradio as gr
 import assets.themes.loadThemes as loadThemes
 from gradio_i18n import Translate
@@ -178,6 +179,7 @@ def download_audio(url, output_dir="ytdl"):
     except Exception as e:
         raise Exception(f"Error extracting audio with yt-dlp: {str(e)}")
 
+@spaces.GPU(duration=60)
 def roformer_separator(audio, model_key, out_format, segment_size, override_seg_size, overlap, batch_size, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     base_name = os.path.splitext(os.path.basename(audio))[0]
     roformer_model = roformer_models[model_key]
@@ -208,7 +210,8 @@ def roformer_separator(audio, model_key, out_format, segment_size, override_seg_
         return stems[1], stems[0]
     except Exception as e:
         raise RuntimeError(f"Roformer separation failed: {e}") from e
-    
+
+@spaces.GPU(duration=60)
 def mdxc_separator(audio, model, out_format, segment_size, override_seg_size, overlap, batch_size, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     base_name = os.path.splitext(os.path.basename(audio))[0]
     try:
@@ -239,6 +242,7 @@ def mdxc_separator(audio, model, out_format, segment_size, override_seg_size, ov
     except Exception as e:
         raise RuntimeError(f"MDX23C separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def mdxnet_separator(audio, model, out_format, hop_length, segment_size, denoise, overlap, batch_size, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     base_name = os.path.splitext(os.path.basename(audio))[0]
     try:
@@ -270,6 +274,7 @@ def mdxnet_separator(audio, model, out_format, hop_length, segment_size, denoise
     except Exception as e:
         raise RuntimeError(f"MDX-NET separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def vrarch_separator(audio, model, out_format, window_size, aggression, tta, post_process, post_process_threshold, high_end_process, batch_size, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     base_name = os.path.splitext(os.path.basename(audio))[0]
     try:
@@ -303,6 +308,7 @@ def vrarch_separator(audio, model, out_format, window_size, aggression, tta, pos
     except Exception as e:
         raise RuntimeError(f"VR ARCH separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def demucs_separator(audio, model, out_format, shifts, segment_size, segments_enabled, overlap, batch_size, norm_thresh, amp_thresh, progress=gr.Progress(track_tqdm=True)):
     base_name = os.path.splitext(os.path.basename(audio))[0]
     try:
@@ -344,6 +350,7 @@ def update_stems(model):
     else:
         return gr.update(visible=False)
 
+@spaces.GPU(duration=60)
 def roformer_batch(path_input, path_output, model_key, out_format, segment_size, override_seg_size, overlap, batch_size, norm_thresh, amp_thresh):
     found_files.clear()
     logs.clear()
@@ -393,6 +400,7 @@ def roformer_batch(path_input, path_output, model_key, out_format, segment_size,
             except Exception as e:
                 raise RuntimeError(f"Roformer batch separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def mdx23c_batch(path_input, path_output, model, out_format, segment_size, override_seg_size, overlap, batch_size, norm_thresh, amp_thresh):
     found_files.clear()
     logs.clear()
@@ -441,6 +449,7 @@ def mdx23c_batch(path_input, path_output, model, out_format, segment_size, overr
             except Exception as e:
                 raise RuntimeError(f"Roformer batch separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def mdxnet_batch(path_input, path_output, model, out_format, hop_length, segment_size, denoise, overlap, batch_size, norm_thresh, amp_thresh):
     found_files.clear()
     logs.clear()
@@ -490,6 +499,7 @@ def mdxnet_batch(path_input, path_output, model, out_format, hop_length, segment
             except Exception as e:
                 raise RuntimeError(f"Roformer batch separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def vrarch_batch(path_input, path_output, model, out_format, window_size, aggression, tta, post_process, post_process_threshold, high_end_process, batch_size, norm_thresh, amp_thresh):
     found_files.clear()
     logs.clear()
@@ -541,6 +551,7 @@ def vrarch_batch(path_input, path_output, model, out_format, window_size, aggres
             except Exception as e:
                 raise RuntimeError(f"Roformer batch separation failed: {e}") from e
 
+@spaces.GPU(duration=60)
 def demucs_batch(path_input, path_output, model, out_format, shifts, segment_size, segments_enabled, overlap, batch_size, norm_thresh, amp_thresh):
     found_files.clear()
     logs.clear()
@@ -592,8 +603,8 @@ def demucs_batch(path_input, path_output, model, out_format, shifts, segment_siz
 with gr.Blocks(theme = loadThemes.load_json() or "NoCrypt/miku", title = "🎵 UVR5 UI 🎵") as app:
     with Translate("assets/languages/translation.yaml", placeholder_langs = ["en", "es", "it", "pt", "ms", "id", "ru", "uk", "th", "zh", "ja", "ko", "tr", "hi"]) as lang:
         gr.Markdown("<h1> 🎵 UVR5 UI 🎵 </h1>")
-        gr.Markdown(_("If you like UVR5 UI you can star my repo on [GitHub](https://github.com/Eddycrack864/UVR5-UI)"))
-        gr.Markdown(_("Try UVR5 UI on Hugging Face with A100 [here](https://huggingface.co/spaces/TheStinger/UVR5_UI)"))
+        gr.Markdown("If you liked this HF Space you can give me a ❤️")
+        gr.Markdown("Try UVR5 UI using Colab [here](https://colab.research.google.com/github/Eddycrack864/UVR5-UI/blob/main/UVR_UI.ipynb)")
         with gr.Tabs():
             with gr.TabItem("BS/Mel Roformer"):
                 with gr.Row():