Update app.py

app.py

@@ -15,6 +15,7 @@ import gradio_client.utils as client_utils
 import validators
 import matchering as mg
 from typing import Any, Optional
+import spaces  # Added for @spaces.GPU
 
 # Logging setup
 logging.basicConfig(level=logging.INFO)
@@ -63,8 +64,9 @@ client_utils._json_schema_to_python_type = patched_json_schema_to_python_type
 # Device and autocast setup
 device = "cuda" if torch.cuda.is_available() else "cpu"
 use_autocast = device == "cuda"
+logger.info(f"Using device: {device}")
 
-# ROFORMER_MODELS and OUTPUT_FORMATS (unchanged)
+# ROFORMER_MODELS and OUTPUT_FORMATS
 ROFORMER_MODELS = {
     "Vocals": {
         'MelBand Roformer | Big Beta 6X by unwa': 'melband_roformer_big_beta6x.ckpt',
@@ -158,7 +160,7 @@ ROFORMER_MODELS = {
 
 OUTPUT_FORMATS = ['wav', 'flac', 'mp3', 'ogg', 'opus', 'm4a', 'aiff', 'ac3']
 
-# Updated CSS with dropdown fixes
+# CSS (unchanged from provided version)
 CSS = """
 /* Modern ve Etkileşimli Tema */
 #app-container {
@@ -173,7 +175,7 @@ CSS = """
     min-height: 100vh;
     background: linear-gradient(135deg, #1a0b2e, #2e1a47);
     position: relative;
-    overflow: visible; /* Changed from hidden to allow dropdowns */
+    overflow: visible;
 }
 body {
     background: none;
@@ -230,7 +232,7 @@ body::after {
     width: 100%;
     transition: transform 0.3s ease;
     position: relative;
-    z-index: 10; /* Ensure container is above other elements */
+    z-index: 10;
 }
 .dubbing-theme:hover {
     transform: translateY(-5px);
@@ -320,7 +322,7 @@ button:hover {
     font-size: 1rem !important;
     transition: border-color 0.3s ease, box-shadow 0.3s ease !important;
     position: relative;
-    z-index: 100; /* High z-index for dropdown */
+    z-index: 100;
 }
 .compact-dropdown:hover {
     border-color: #ff8787 !important;
@@ -342,11 +344,11 @@ button:hover {
     border: 2px solid #ff6b6b !important;
     border-radius: 8px !important;
     color: #e0e0e0 !important;
-    max-height: 300px !important; /* Increased for more options */
+    max-height: 300px !important;
     overflow-y: auto !important;
-    z-index: 200 !important; /* Ensure menu is above other elements */
+    z-index: 200 !important;
     width: 100% !important;
-    opacity: 1 !important; /* Always visible */
+    opacity: 1 !important;
     visibility: visible !important;
     position: absolute !important;
     top: 100% !important;
@@ -409,7 +411,7 @@ button:hover {
         max-width: 100%;
     }
     .compact-dropdown .gr-dropdown-menu {
-        max-height: 200px !important; /* Smaller for mobile */
+        max-height: 200px !important;
     }
 }
 """
@@ -453,7 +455,7 @@ def download_audio(url, cookies_file=None, out_dir="ytdl"):
         logger.error(f"Download failed: {e}")
         raise RuntimeError(f"Download failed: {e}")
 
-
+@spaces.GPU
 def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, pitch_shift, model_dir, output_dir, out_format, norm_thresh, amp_thresh, batch_size, exclude_stems="", progress=gr.Progress(track_tqdm=True)):
     """Separate audio into stems using a Roformer model."""
     if not audio:
@@ -473,7 +475,7 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
     else:
         raise ValueError(f"Model '{model_key}' not found.")
     
-    logger.info(f"Separating {base_name} with {model_key}")
+    logger.info(f"Separating {base_name} with {model_key} on {device}")
     try:
         separator = Separator(
             log_level=logging.INFO,
@@ -500,6 +502,7 @@ def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, p
         logger.error(f"Separation failed: {e}")
         raise RuntimeError(f"Separation failed: {e}")
 
+@spaces.GPU
 def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_tta, model_dir, output_dir, norm_thresh, amp_thresh, batch_size, ensemble_method, exclude_stems="", weights_str="", progress=gr.Progress()):
     """Perform ensemble processing on audio using multiple Roformer models."""
     if not audio or not model_keys:
@@ -512,7 +515,7 @@ def auto_ensemble_process(audio, model_keys, seg_size, overlap, out_format, use_
     os.makedirs(output_dir, exist_ok=True)
     
     base_name = os.path.splitext(os.path.basename(audio))[0]
-    logger.info(f"Ensemble for {base_name} with {model_keys}")
+    logger.info(f"Ensemble for {base_name} with {model_keys} on {device}")
     
     all_stems = []
     total_models = len(model_keys)