Update app.py

app.py

@@ -7,6 +7,11 @@ between: the reconstructed audio, the waveform comparison, the latent
 representation itself, and the numbers behind the compression.
 """
 
+# `spaces` must be imported before torch (or anything that touches CUDA) so
+# its ZeroGPU shims are installed first. This Space runs on ZeroGPU hardware,
+# which requires at least one @spaces.GPU-decorated function.
+import spaces
+
 import os
 import time
 import tempfile
@@ -145,6 +150,7 @@ def make_latent_figure(latent_matrix: np.ndarray):
 # Core inference
 # --------------------------------------------------------------------------
 
+@spaces.GPU(duration=60)
 def process(audio_path, progress=gr.Progress(track_tqdm=False)):
     if audio_path is None:
         raise gr.Error("Upload or record a clip first.")
@@ -295,7 +301,7 @@ CUSTOM_CSS = f"""
 }}
 """
 
-with gr.Blocks(css=CUSTOM_CSS, title="BlueCodec — speech autoencoder", theme=gr.themes.Base()) as demo:
+with gr.Blocks(title="BlueCodec — speech autoencoder") as demo:
     gr.Markdown(
         "# BlueCodec\n"
         "Upload a voice clip and pass it through a 24-dimensional latent "
@@ -356,4 +362,4 @@ with gr.Blocks(css=CUSTOM_CSS, title="BlueCodec — speech autoencoder", theme=g
     )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch(theme=gr.themes.Base(), css=CUSTOM_CSS)