update code

.idea/workspace.xml

@@ -6,7 +6,7 @@
   <component name="ChangeListManager">
     <list default="true" id="23565123-73ab-4f40-a9ef-1086e0c9e1ec" name="Changes" comment="">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/src/audio_utils.py" beforeDir="false" afterPath="$PROJECT_DIR$/src/audio_utils.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/script.py" beforeDir="false" afterPath="$PROJECT_DIR$/script.py" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -62,6 +62,7 @@
       <option name="presentableId" value="Default" />
       <updated>1742573353560</updated>
       <workItem from="1742573355153" duration="587000" />
+      <workItem from="1742806974298" duration="157000" />
     </task>
     <servers />
   </component>

script.py

@@ -1,16 +1,21 @@
 import io
 import time
+import os
 
 import torch
 import tqdm.auto as tqdm
 from datasets import load_dataset
 
+import numpy as np
+import pandas as pd
+
 from models import Model
 from preprocess import preprocess
 # from pathlib import Path
 
 # from src.rawnet_model import RawNet
-from src.utils import *
+# from src.utils import *
+
 
 # Import your model and anything else you want
 # You can even install other packages included in your repo
@@ -29,7 +34,7 @@ dataset_remote = load_dataset(DATASET_PATH,split = "test",streaming = True)
 # load your model
 device = "cuda:0"
 
-## DEFINE RAWNET2 MODEL
+# # DEFINE RAWNET2 MODEL
 # this_folder = Path(__file__).parent
 # config_path = this_folder / 'config' / 'rawnet_config.yaml'
 # config = read_yaml(config_path)