fix!: adapt the code to use Zero GPU

app.py

@@ -9,6 +9,7 @@ import traceback
 import logging
 from inference import proc_folder_direct
 from pathlib import Path
+from transformers import pipeline
 
 OUTPUT_FOLDER = "separation_results/"
 INPUT_FOLDER = "input"
@@ -150,10 +151,11 @@ def delete_folders_and_files(input_dir):
 
     print("Cleanup completed.")
 
+@spaces.GPU(duration=300)  # Adjust the duration as needed
 def process_audio(uploaded_file):
     try:
         yield "Processing audio...", None
-        
+
         if uploaded_file:
             input_path, formatted_title = handle_file_upload(uploaded_file)
             if input_path is None:
@@ -166,7 +168,7 @@ def process_audio(uploaded_file):
 
         yield "Starting Mel Band Roformer inference...", None
         proc_folder_direct("mel_band_roformer", "configs/config_mel_band_roformer_vocals.yaml", "results/model_mel_band_roformer_vocals.ckpt", f"{INPUT_FOLDER}/wav", OUTPUT_FOLDER, extract_instrumental=True)
-        
+
         yield "Starting HTDemucs inference...", None
         proc_folder_direct("htdemucs", "configs/config_htdemucs_bass.yaml", "results/model_htdemucs_bass.th", f"{INPUT_FOLDER}/wav", OUTPUT_FOLDER)
 
@@ -220,5 +222,5 @@ with gr.Blocks() as demo:
         outputs=[log_output, processed_audio_output],
         show_progress=True
     )
-        
+
 demo.launch()