Upload 2 files

app.py

@@ -103,8 +103,12 @@ def detect_on_gpu(dataset):
 
 # 修改音频伪造检测主函数
 def audio_deepfake_detection(demonstrations, query_audio_path):
-    demonstration_paths = [audio[0] for audio in demonstrations if audio[0] is not None]
-    demonstration_labels = [audio[1] for audio in demonstrations if audio[1] is not None]
+    demonstration_paths = []
+    demonstration_labels = []
+    for demonstration in demonstrations:
+        if demonstration[0] is not None:
+            demonstration_paths.append(demonstration[0])
+            demonstration_labels.append(demonstration[1])
     if len(demonstration_paths) != len(demonstration_labels):
         demonstration_labels = demonstration_labels[:len(demonstration_paths)]
     

dataset.py

@@ -6,7 +6,7 @@ import librosa
 import numpy as np
 
 class DemoDataset(Dataset):
-    def __init__(self, demonstration_paths, demonstration_labels, query_path, sample_rate=16000):
+    def __init__(self, demonstration_paths=[], demonstration_labels=[], query_path="", sample_rate=16000):
         self.sample_rate = sample_rate
         self.query_path = query_path