import models
import os
import pandas as pd
import numpy as np
from run_model import run
import soundfile as sf

folder = '/nfs/NAS6/SABIOD/SITE/KM3Net/DATA_WAV/'
fns = []
for f in os.listdir(folder):
    if os.path.isdir(folder+f) and f > 'RUN7700':
        fns.extend(f+'/'+pd.Series(os.listdir(folder+f)))

model = models.get['stft_depthwise_ksize'](128, 7)
stdc='models/stft_depthwise_ovs_128_k7_r1.stdc'
preds = run(fns, stdc, model, folder, pool=False, batch_size=64, fe=50000, lensample=120)
preds.to_pickle('KM3Net_'+stdc.split('/')[-1].rsplit('.',1)[0]+'.preds')