minor fix

train.py

@@ -15,7 +15,7 @@ def main():
     labels = []
     with open(path / "train.csv", "r") as file:
         for line in file:
-            labels.append(line.strip().split(","))
+            labels.append(line.strip().split(",")[0])
 
     spark = SparkSession.builder.appName("train").getOrCreate()
     pipe = SpectrogramPipe(spark)