Update pipeline.py
Browse files- pipeline.py +2 -5
pipeline.py
CHANGED
|
@@ -39,16 +39,13 @@ efficientnet_model = tf.keras.layers.TFSMLayer(
|
|
| 39 |
# Audio Ensemble: 3 models vote — majority wins
|
| 40 |
# ─────────────────────────────────────────────────────────────────────────────
|
| 41 |
AUDIO_MODELS = [
|
| 42 |
-
"MelodyMachine/Deepfake-audio-detection-V2",
|
| 43 |
-
"MelodyMachine/Deepfake-audio-detection",
|
| 44 |
"Gustking/wav2vec2-large-xlsr-deepfake-audio-classification",
|
| 45 |
]
|
| 46 |
AUDIO_SAMPLE_RATE = 16000
|
| 47 |
|
| 48 |
# ─── Thresholds ───────────────────────────────────────────────────────────────
|
| 49 |
-
REAL_THRESHOLD = 0.
|
| 50 |
-
FAKE_THRESHOLD = 0.
|
| 51 |
-
|
| 52 |
print("Loading audio ensemble models ...")
|
| 53 |
ensemble = []
|
| 54 |
for model_id in AUDIO_MODELS:
|
|
|
|
| 39 |
# Audio Ensemble: 3 models vote — majority wins
|
| 40 |
# ─────────────────────────────────────────────────────────────────────────────
|
| 41 |
AUDIO_MODELS = [
|
|
|
|
|
|
|
| 42 |
"Gustking/wav2vec2-large-xlsr-deepfake-audio-classification",
|
| 43 |
]
|
| 44 |
AUDIO_SAMPLE_RATE = 16000
|
| 45 |
|
| 46 |
# ─── Thresholds ───────────────────────────────────────────────────────────────
|
| 47 |
+
REAL_THRESHOLD = 0.55
|
| 48 |
+
FAKE_THRESHOLD = 0.70
|
|
|
|
| 49 |
print("Loading audio ensemble models ...")
|
| 50 |
ensemble = []
|
| 51 |
for model_id in AUDIO_MODELS:
|