Update Main.py

Main.py

@@ -17,6 +17,7 @@ from tqdm import tqdm
 import time
 
 import numpy as np
+import os
 
 warnings.filterwarnings("ignore")
 
@@ -33,7 +34,9 @@ main_path = data['MainPath']
 device = data['Device']
 diag_attn = data['DiagAttn']
 
-BestModelPath = '/home/siddc/SiddC/Misc/WatchMeSpeak/WatchMeSpeak/Best_GlobalModel_500_0_0.pt'
+dir_path = os.path.dirname(os.path.realpath(__file__))
+
+BestModelPath = dir_path + '/Best_GlobalModel_500_0_0.pt'
 
 def pre_process_mfcc(mfcc):
     
@@ -47,7 +50,7 @@ def pre_process_mfcc(mfcc):
 
 def wav2art(wav):
     rate = 16000
-    mfcc = librosa.feature.mfcc(wav, 16000, n_mfcc=13, hop_length=int(0.010*rate), n_fft=int(0.020*rate))
+    mfcc = librosa.feature.mfcc(y = wav, sr = 16000, n_mfcc=13, hop_length=int(0.010*rate), n_fft=int(0.020*rate))
     mfcc = pre_process_mfcc(mfcc)
 
     mfcc = torch.tensor([mfcc]).float()