COnfig fix

app.py

@@ -1,12 +1,13 @@
 import argparse
 import os
+import json
 
 import wget
 import torch
 import torchaudio
 import gradio as gr
 
-from src.training.dcc_tf import Net as Waveformer
+from dcc_tf import Net as Waveformer
 
 TARGETS = [
     "Acoustic_guitar", "Applause", "Bark", "Bass_drum",
@@ -31,9 +32,11 @@ if not os.path.exists('default_ckpt.pt'):
     wget.download(ckpt_url)
 
 # Instantiate model
-params = utils.Params('default_config.json')
-model = Waveformer(**params.model_params)
-model.load_state_dict(torch.load('default_ckpt.pt', map_location=torch.device('cpu')))
+with open('default_config.json') as f:
+    params = json.load(f)
+model = Waveformer(**params['model_params'])
+model.load_state_dict(
+    torch.load('default_ckpt.pt', map_location=torch.device('cpu'))['model_state_dict'])
 model.eval()
 
 def waveformer(audio, label_choices):