suff

app.py

@@ -16,13 +16,7 @@ import gradio as gr
 from vampnet.interface import Interface, signal_concat
 from vampnet import mask as pmask
 
-if torch.cuda.is_available():
-    device = "cuda"
-elif torch.backends.mps.is_available():
-    device = "mps"
-else:
-    device = "cpu"
-
+device="cpu"
 print(f"using device {device}\n"*10)
 
 interface = Interface.default()
@@ -137,6 +131,13 @@ def _vamp_internal(
         typical_mass, typical_min_tokens, top_p, 
         sample_cutoff, stretch_factor, sampling_steps, beat_mask_ms, num_feedback_steps, api=False, harp=False
     ):
+    if torch.cuda.is_available():
+        device = "cuda"
+    elif torch.backends.mps.is_available():
+        device = "mps"
+    else:
+        device = "cpu"
+
 
     print("args!")
     print(f"seed: {seed}")