Update dune_extraction.py

dune_extraction.py

@@ -10,12 +10,9 @@ from bisect import bisect_left
 from concurrent.futures import ThreadPoolExecutor
 from pathlib import Path
 
-sys.path.insert(0, "/home/ubuntu/server1/zs_cleaning/layla")
-sys.path.insert(1, "/home/ubuntu/zs_cleaning/layla")
 
-from src_takane_nano.codec.audio_processing.kanadec_audio_tokenizer import (
-     load_nemo_nano_audio_tokenizer,
-)
+from codec.audio_processing.dune_codec import load_dune_audio_tokenizer
+
 
 import librosa
 import numpy as np
@@ -1516,7 +1513,7 @@ def main():
      device = "cuda"
      print(f"Loading Kanadec nano audio tokenizer on {device}...")
 
-     codec_model = load_nemo_nano_audio_tokenizer(
+     codec_model = load_dune_audio_tokenizer(
           MODEL_ID,
           device=device,
      )