waidhoferj commited on
Commit
6e258bf
1 Parent(s): 81caeb5

updated config path

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -13,8 +13,7 @@ from torch import nn
13
  import yaml
14
  import torchaudio
15
 
16
- CONFIG_FILE = Path("models/config/production.yaml")
17
- MODEL_CLS = ResidualDancer
18
 
19
  DANCE_MAPPING_FILE = Path("data/dance_mapping.csv")
20
 
 
13
  import yaml
14
  import torchaudio
15
 
16
+ CONFIG_FILE = Path("models/weights/ResidualDancer/multilabel/config.yaml")
 
17
 
18
  DANCE_MAPPING_FILE = Path("data/dance_mapping.csv")
19