update for zero-gpu

app.py

@@ -6,23 +6,11 @@ from huggingface_hub import hf_hub_download
 import numpy as np
 import sphn
 import torch
+import spaces
 
 from moshi.models import loaders
 
 
-def seed_all(seed):
-    torch.manual_seed(seed)
-    if torch.cuda.is_available():
-        torch.cuda.manual_seed(seed)
-        torch.cuda.manual_seed_all(seed)  # for multi-GPU setups
-    random.seed(seed)
-    np.random.seed(seed)
-    torch.backends.cudnn.deterministic = True
-    torch.backends.cudnn.benchmark = False
-
-
-seed_all(42424242)
-
 device = "cuda" if torch.cuda.device_count() else "cpu"
 num_codebooks = 32
 
@@ -34,6 +22,7 @@ mimi.eval()
 print("mimi loaded")
 
 
+@spaces.GPU
 def mimi_streaming_test(input_wave, max_duration_sec=10.0):
     pcm_chunk_size = int(mimi.sample_rate / mimi.frame_rate)
     # wget https://github.com/metavoiceio/metavoice-src/raw/main/assets/bria.mp3