Update app.py

app.py

@@ -1,14 +1,10 @@
 import gradio as gr
-import spaces
 import torch
 import dac
 import io
 from audiotools import AudioSignal
 from pydub import AudioSegment
 
-# Ensure we're using CPU even if GPU is available
-torch.set_default_tensor_type(torch.FloatTensor)
-
 class DACApi:
     def __init__(self, model_type="44khz", model_bitrate="16kbps"):
         self.model_type = model_type
@@ -16,8 +12,10 @@ class DACApi:
         self.model_path = dac.utils.download(model_type, model_bitrate)
         print("Loading DAC model...")
         self.model = dac.DAC.load(self.model_path)
-        self.model.to('cpu')
+        self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        self.model.to(self.device)
 
+    @spaces.GPU
     def encode_audio(self, audio):
         # Convert audio to WAV
         audio = AudioSegment.from_file(audio.name)
@@ -27,6 +25,7 @@ class DACApi:
 
         # Load audio signal
         signal = AudioSignal(wav_io)
+        signal = signal.to(self.device)
 
         # Compress audio
         print("Compressing audio...")
@@ -39,9 +38,11 @@ class DACApi:
         
         return output
 
+    @spaces.GPU
     def decode_audio(self, compressed_file):
         # Load compressed audio
         compressed = dac.DACFile.load(compressed_file)
+        compressed = compressed.to(self.device)
 
         # Decompress audio
         print("Decompressing audio...")
@@ -56,12 +57,10 @@ class DACApi:
 
 dac_api = DACApi()
 
-@spaces.CPU
 def encode(audio):
     compressed = dac_api.encode_audio(audio)
     return compressed
 
-@spaces.CPU
 def decode(compressed_file):
     decompressed = dac_api.decode_audio(compressed_file)
     return decompressed
@@ -84,4 +83,4 @@ with gr.Blocks() as demo:
         
         decode_button.click(decode, inputs=compressed_input, outputs=decoded_output)
 
-demo.launch()
+demo.launch()