Update app.py

app.py

@@ -1,10 +1,8 @@
 import re
-
 import gradio as gr
 import numpy as np
 import tempfile
 from einops import rearrange
-#from vocos import Vocos
 from pydub import AudioSegment, silence
 from model import CFM, UNetT, DiT, MMDiT
 from cached_path import cached_path
@@ -25,19 +23,24 @@ from transformers import pipeline
 import click
 import soundfile as sf
 
+# Import 'spaces' early to avoid CUDA initialization conflicts
 try:
     import spaces
     USING_SPACES = True
 except ImportError:
     USING_SPACES = False
 
+# Import PyTorch and related libraries after 'spaces'
+import torch
+import torchaudio
+
+# GPU decorator handling for 'spaces'
 def gpu_decorator(func):
     if USING_SPACES:
         return spaces.GPU(func)
     else:
         return func
-import torch
-import torchaudio
+
 # Determine the device
 device = (
     "cuda"
@@ -55,7 +58,6 @@ else:
 
 # Create the torch.device object
 device = torch.device(device)
-
 print(f"Using device: {device}, dtype: {dtype}")
 
 pipe = pipeline(