track time

script.py

@@ -96,6 +96,8 @@ print('Loaded Weights')
 # del model
 # model = Model().to(device)
 
+SAMPLING_RATES = [8000, 16000, 22050, 24000, 32000, 44100, 48000, "other"]
+
 # iterate over the dataset
 out = []
 for el in tqdm.tqdm(dataset_remote):
@@ -140,13 +142,12 @@ for el in tqdm.tqdm(dataset_remote):
     # RUNNING ON HUGGINGFACE
     # total_time = time.time() - start_time
 
-    if sr == 16000:
-        freq_factor = 50
-    elif sr > 16000:
-        freq_factor = 70
-    elif sr < 16000:
-        freq_factor = 30
-    total_time = freq_factor
+    if sr in SAMPLING_RATES:
+        index = SAMPLING_RATES.index(sr)
+    else:
+        index = SAMPLING_RATES.index("other")
+
+    total_time = 1 << index
     # total_time = 0.0001 + freq_factor
 
     out.append(dict(id=el["id"], pred=pred, score=score, time=total_time))