Serhiy Stetskovych commited on
Commit
e42e488
1 Parent(s): a60a79a

Fix configs

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,10 +45,10 @@ texts
45
 
46
 
47
  apollo_config = get_config('configs/apollo.yaml')
48
- apollo_vocal_config = get_config('configs/config_apollo_vocal.yaml')
49
  apollo_model = look2hear.models.BaseModel.from_pretrain('weights/apollo.bin', **apollo_config['model']).to(device)
50
- apollo_vocal = look2hear.models.BaseModel.from_pretrain('weights/apollo_vocal.bin', **apollo_vocal_config['model']).to(device)
51
- apollo_vocal2 = look2hear.models.BaseModel.from_pretrain('weights/apollo_vocal2.bin', **apollo_vocal_config['model']).to(device)
52
 
53
 
54
 
 
45
 
46
 
47
  apollo_config = get_config('configs/apollo.yaml')
48
+ apollo_vocal2_config = get_config('configs/config_apollo_vocal.yaml')
49
  apollo_model = look2hear.models.BaseModel.from_pretrain('weights/apollo.bin', **apollo_config['model']).to(device)
50
+ apollo_vocal = look2hear.models.BaseModel.from_pretrain('weights/apollo_vocal.bin', **apollo_config['model']).to(device)
51
+ apollo_vocal2 = look2hear.models.BaseModel.from_pretrain('weights/apollo_vocal2.bin', **apollo_vocal2_config['model']).to(device)
52
 
53
 
54