codec

app.py

@@ -9,6 +9,7 @@ from dataclasses import dataclass, asdict
 import numpy as np
 import audiotools as at
 import argbind
+import torch
 
 import gradio as gr
 from vampnet.interface import Interface
@@ -17,10 +18,15 @@ from vampnet import mask as pmask
 Interface = argbind.bind(Interface)
 # AudioLoader = argbind.bind(at.data.datasets.AudioLoader)
 
-conf = argbind.parse_args()
+with open("conf/interface.yml") as f:
+    conf = yaml.safe_load(f)
+
 
 with argbind.scope(conf):
-    interface = Interface()
+    interface = Interface(
+        device="cuda" if torch.cuda.is_available() else "cpu",
+    )
+
     # loader = AudioLoader()
     print(f"interface device is {interface.device}")