debug

app.py

@@ -15,11 +15,21 @@ from generate_audio import (
 )  
 import uuid
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-vq_model = RQBottleneckTransformer.load_model(
-        "whisper-vq-stoks-medium-en+pl-fixed.model"
-    ).to(device)
-vq_model.ensure_whisper(device)
+
+@spaces.GPU
+def vq_model_load():
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    vq_model = RQBottleneckTransformer.load_model(
+            "whisper-vq-stoks-medium-en+pl-fixed.model"
+        ).to(device)
+    vq_model.ensure_whisper(device)
+    return vq_model
+# device = "cuda" if torch.cuda.is_available() else "cpu"
+# vq_model = RQBottleneckTransformer.load_model(
+#         "whisper-vq-stoks-medium-en+pl-fixed.model"
+#     ).to(device)
+# vq_model.ensure_whisper(device)
+vq_model = vq_model_load()
 
 @spaces.GPU
 def audio_to_sound_tokens_whisperspeech(audio_path):