Upload app.py

app.py

@@ -8,6 +8,7 @@ import json
 import os
 import traceback
 import gc
+import torch
 
 # Import your modules
 from engine import compute_mapss_measures
@@ -99,6 +100,9 @@ def process_audio_files(zip_file, model_name, layer, alpha):
                 }
                 layer_final = layer if layer is not None else model_defaults.get(model_name, 12)
             
+            # Check if GPU is available
+            max_gpus = 1 if torch.cuda.is_available() else 0
+            
             # Run experiment with compute_mapss_measures
             results_dir = compute_mapss_measures(
                 models=[model_name],
@@ -106,7 +110,7 @@ def process_audio_files(zip_file, model_name, layer, alpha):
                 layer=layer_final,
                 alpha=alpha,
                 verbose=True,
-                max_gpus=1,  # Limit to 1 GPU for HF Space
+                max_gpus=max_gpus,  # Use GPU if available, otherwise CPU
                 add_ci=False  # Disable CI for faster processing
             )
             
@@ -317,9 +321,6 @@ def create_interface():
                     max_lines=10
                 )
         
-        # gr.Markdown("""
-        # """)
-        
         # Set up the processing
         process_btn.click(
             fn=process_audio_files,