Upload folder using huggingface_hub

inference.py

@@ -7,7 +7,6 @@ import torch
 from typing import List, Tuple, Optional
 import pytorch_lightning as pl
 from model import MusicClassifier, MusicAudioClassifier
-import argparse
 import torch
 import torchaudio
 import scipy.signal as signal

preprocess.py

@@ -7,7 +7,6 @@ from collections import Counter
 import os
 import argparse
 from tqdm import tqdm
-import shutil
 import concurrent.futures
 
 
@@ -41,6 +40,7 @@ def find_optimal_segment_length(downbeats, round_decimal=1, bar_length = 4):
     most_common_interval = interval_counter.most_common(1)[0][0]
     
     # 정제된 downbeat 위치 찾기
+
     cleaned_downbeats = [downbeats[0]]  # 첫 번째 위치는 항상 포함
     
     for i in range(1, len(downbeats)):

requirements.txt

@@ -4,4 +4,8 @@ tqdm
 imblearn
 transformers
 https://github.com/CPJKU/beat_this/archive/main.zip
-pytorch_lightning
+pytorch_lightning
+scikit-learn
+torchaudio
+deepspeed
+librosa