update script.py

.gitignore

@@ -1,6 +1,8 @@
 # Ignore weights
 checkpoints/RAWNET_ASVSPOOF_FOR_INTHEWILD_PURDUE.pth
 
+.idea
+
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.9" />
+  </component>
   <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9" project-jdk-type="Python SDK" />
 </project>

.idea/workspace.xml

@@ -5,8 +5,9 @@
   </component>
   <component name="ChangeListManager">
     <list default="true" id="23565123-73ab-4f40-a9ef-1086e0c9e1ec" name="Changes" comment="">
-      <change beforePath="$PROJECT_DIR$/README.md" beforeDir="false" afterPath="$PROJECT_DIR$/README.md" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/test.py" beforeDir="false" />
+      <change beforePath="$PROJECT_DIR$/.idea/misc.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/misc.xml" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" 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" />
@@ -30,29 +31,29 @@
   <component name="MarkdownSettingsMigration">
     <option name="stateVersion" value="1" />
   </component>
-  <component name="ProjectColorInfo"><![CDATA[{
-  "associatedIndex": 8
-}]]></component>
+  <component name="ProjectColorInfo">{
+  &quot;associatedIndex&quot;: 8
+}</component>
   <component name="ProjectId" id="2udL9TBw08aLOM9GtJKtPFHLCAg" />
   <component name="ProjectViewState">
     <option name="hideEmptyMiddlePackages" value="true" />
     <option name="showLibraryContents" value="true" />
   </component>
-  <component name="PropertiesComponent"><![CDATA[{
-  "keyToString": {
-    "RunOnceActivity.OpenProjectViewOnStart": "true",
-    "RunOnceActivity.ShowReadmeOnStart": "true",
-    "WebServerToolWindowFactoryState": "true",
-    "last_opened_file_path": "/Users/davidesalvi/Desktop/PHD/projects/polimi-ispl/ispl_safe",
-    "node.js.detected.package.eslint": "true",
-    "node.js.detected.package.tslint": "true",
-    "node.js.selected.package.eslint": "(autodetect)",
-    "node.js.selected.package.tslint": "(autodetect)",
-    "nodejs_package_manager_path": "npm",
-    "settings.editor.selected.configurable": "web.server",
-    "vue.rearranger.settings.migration": "true"
+  <component name="PropertiesComponent">{
+  &quot;keyToString&quot;: {
+    &quot;RunOnceActivity.OpenProjectViewOnStart&quot;: &quot;true&quot;,
+    &quot;RunOnceActivity.ShowReadmeOnStart&quot;: &quot;true&quot;,
+    &quot;WebServerToolWindowFactoryState&quot;: &quot;true&quot;,
+    &quot;last_opened_file_path&quot;: &quot;/Users/davidesalvi/Desktop/PHD/projects/polimi-ispl/ispl_safe&quot;,
+    &quot;node.js.detected.package.eslint&quot;: &quot;true&quot;,
+    &quot;node.js.detected.package.tslint&quot;: &quot;true&quot;,
+    &quot;node.js.selected.package.eslint&quot;: &quot;(autodetect)&quot;,
+    &quot;node.js.selected.package.tslint&quot;: &quot;(autodetect)&quot;,
+    &quot;nodejs_package_manager_path&quot;: &quot;npm&quot;,
+    &quot;settings.editor.selected.configurable&quot;: &quot;web.server&quot;,
+    &quot;vue.rearranger.settings.migration&quot;: &quot;true&quot;
   }
-}]]></component>
+}</component>
   <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
   <component name="TaskManager">
     <task active="true" id="Default" summary="Default task">

script.py

@@ -1,22 +1,24 @@
-import pandas as pd
-from datasets import load_dataset
-import numpy as np
-import tqdm.auto as tqdm
-import os
 import io
-import torch
 import time
 
+import torch
+import tqdm.auto as tqdm
+from datasets import load_dataset
+
+from models import Model
+from preprocess import preprocess
+from pathlib import Path
+
+from src.rawnet_model import RawNet
+from src.utils import *
+
 # Import your model and anything else you want
 # You can even install other packages included in your repo
-# However, during the evaluation the container will not have access to the internet. 
+# However, during the evaluation the container will not have access to the internet.
 # So you must include everything you need in your model repo. Common python libraries will be installed.
 # Feel free to contact us to add dependencies to the requiremnts.txt
-# For testing, this is the docker image that will be used https://github.com/huggingface/competitions/blob/main/Dockerfile 
-# It can be pulled here https://hub.docker.com/r/huggingface/competitions/tags 
-
-from models import Model
-from preprocess import preprocess
+# For testing, this is the docker image that will be used https://github.com/huggingface/competitions/blob/main/Dockerfile
+# It can be pulled here https://hub.docker.com/r/huggingface/competitions/tags
 
 
 # load the dataset. dataset will be automatically downloaded to /tmp/data during evaluation
@@ -26,8 +28,18 @@ dataset_remote = load_dataset(DATASET_PATH,split = "test",streaming = True)
 
 # load your model
 device = "cuda:0"
-model = Model().to(device)
 
+# DEFINE RAWNET2 MODEL
+this_folder = Path(__file__).parent
+config_path = this_folder / 'config' / 'rawnet_config.yaml'
+config = read_yaml(config_path)
+model = RawNet(config['model'], device)
+
+model_path = this_folder / 'checkpoints' / 'RAWNET_ASVSPOOF_FOR_INTHEWILD_PURDUE.pth'
+model.load_state_dict(torch.load(model_path, map_location=device))
+
+# Define old model
+model = Model().to(device)
 
 # iterate over the dataset
 out = []