Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 import torchaudio
@@ -9,7 +10,7 @@ import numpy as np
 from pydub import AudioSegment
 
 # Load model and configuration
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = 'cuda'
 
 dit_checkpoint_path, dit_config_path = load_custom_model_from_hf("Plachta/Seed-VC",
                                                 "DiT_seed_v2_uvit_whisper_small_wavenet_bigvgan_pruned.pth",
@@ -128,6 +129,7 @@ def crossfade(chunk1, chunk2, overlap):
 overlap_frame_len = 16
 bitrate = "320k"
 
+@spaces.GPU
 @torch.no_grad()
 @torch.inference_mode()
 def voice_conversion(source, target, diffusion_steps, length_adjust, inference_cfg_rate, f0_condition, auto_f0_adjust, pitch_shift):