Update pipeline.py

pipeline.py

@@ -174,9 +174,9 @@ def load_audio_model():
     
     model = RawNet(d_args = d_args, device='cpu')
 
-    #Load pretrained ckpt.
+    #Load ckpt.
     model_dict = model.state_dict()
-    ckpt = torch.load('pre_trained_DF_RawNet2.pth', map_location=torch.device('cpu'))
+    ckpt = torch.load('RawNet2.pth', map_location=torch.device('cpu'))
     model.load_state_dict(ckpt, model_dict)
     return model