Upload 2 files

app.py

@@ -9,6 +9,7 @@ import os
 import traceback
 import gc
 import torch
+import spaces  # Import spaces for ZeroGPU
 
 # Import your modules
 from engine import compute_mapss_measures
@@ -16,6 +17,7 @@ from models import get_model_config, cleanup_all_models
 from config import DEFAULT_ALPHA
 from utils import clear_gpu_memory
 
+@spaces.GPU(duration=300)  # Request GPU for up to 5 minutes
 def process_audio_files(zip_file, model_name, layer, alpha):
     """
     Process uploaded ZIP file containing audio mixtures.

requirements.txt

@@ -1,3 +1,6 @@
+# ZeroGPU specific
+spaces>=0.19.0
+
 # Core dependencies
 gradio>=4.0.0
 torch>=2.0.0
@@ -19,8 +22,8 @@ pandas>=2.0.0
 safetensors>=0.4.0
 sentencepiece>=0.1.99  # For some tokenizers
 
-# Optional optimizations
-triton>=2.1.0  # For faster attention if available
+# Optional optimizations (comment out if causing issues)
+# triton>=2.1.0  # For faster attention if available
 
 # Memory management
 psutil>=5.9.0