Hugo Flores Garcia commited on
Commit
bc21614
1 Parent(s): b566033
Files changed (1) hide show
  1. app.py +9 -11
app.py CHANGED
@@ -15,20 +15,18 @@ import gradio as gr
15
  from vampnet.interface import Interface
16
  from vampnet import mask as pmask
17
 
18
- Interface = argbind.bind(Interface)
19
  # AudioLoader = argbind.bind(at.data.datasets.AudioLoader)
20
 
21
- with open("conf/interface.yml") as f:
22
- conf = yaml.safe_load(f)
 
 
 
 
23
 
24
-
25
- with argbind.scope(conf):
26
- interface = Interface(
27
- device="cuda" if torch.cuda.is_available() else "cpu",
28
- )
29
-
30
- # loader = AudioLoader()
31
- print(f"interface device is {interface.device}")
32
 
33
  # dataset = at.data.datasets.AudioDataset(
34
  # loader,
 
15
  from vampnet.interface import Interface
16
  from vampnet import mask as pmask
17
 
18
+ # Interface = argbind.bind(Interface)
19
  # AudioLoader = argbind.bind(at.data.datasets.AudioLoader)
20
 
21
+ interface = Interface(
22
+ coarse_ckpt="./models/vampnet/coarse.pth",
23
+ coarse2fine_ckpt="./models/vampnet/c2f.pth",
24
+ codec_ckpt="./models/vampnet/codec.pth",
25
+ device="cuda" if torch.cuda.is_available() else "cpu",
26
+ )
27
 
28
+ # loader = AudioLoader()
29
+ print(f"interface device is {interface.device}")
 
 
 
 
 
 
30
 
31
  # dataset = at.data.datasets.AudioDataset(
32
  # loader,