Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -32,7 +32,7 @@ efficientnet_model = tf.keras.layers.TFSMLayer(
|
|
| 32 |
# Wav2Vec2 fine-tuned for real vs fake audio classification
|
| 33 |
# Accuracy: 98.82% on evaluation set
|
| 34 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 35 |
-
AUDIO_MODEL_ID = "
|
| 36 |
AUDIO_SAMPLE_RATE = 16000
|
| 37 |
|
| 38 |
print(f"Loading audio model: {AUDIO_MODEL_ID} ...")
|
|
|
|
| 32 |
# Wav2Vec2 fine-tuned for real vs fake audio classification
|
| 33 |
# Accuracy: 98.82% on evaluation set
|
| 34 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
| 35 |
+
AUDIO_MODEL_ID = "Gustking/wav2vec2-large-xlsr-deepfake-audio-classification"
|
| 36 |
AUDIO_SAMPLE_RATE = 16000
|
| 37 |
|
| 38 |
print(f"Loading audio model: {AUDIO_MODEL_ID} ...")
|