Update app.py
Browse files
app.py
CHANGED
|
@@ -28,9 +28,9 @@ class HuggingFaceFeatureExtractor:
|
|
| 28 |
|
| 29 |
FEATURE_EXTRACTORS = {
|
| 30 |
"wavlm-base": lambda: HuggingFaceFeatureExtractor(WavLMModel, "microsoft/wavlm-base"),
|
| 31 |
-
"wavLM-V1": lambda: HuggingFaceFeatureExtractor(WavLMModel, "DavidCombei/wavLM-base-
|
| 32 |
-
"wavLM-V2": lambda: HuggingFaceFeatureExtractor(WavLMModel, "DavidCombei/wavLM-base-
|
| 33 |
-
"wavLM-V3": lambda: HuggingFaceFeatureExtractor(WavLMModel, "DavidCombei/wavLM-base-
|
| 34 |
}
|
| 35 |
|
| 36 |
model1 = joblib.load('model1.joblib')
|
|
|
|
| 28 |
|
| 29 |
FEATURE_EXTRACTORS = {
|
| 30 |
"wavlm-base": lambda: HuggingFaceFeatureExtractor(WavLMModel, "microsoft/wavlm-base"),
|
| 31 |
+
"wavLM-V1": lambda: HuggingFaceFeatureExtractor(WavLMModel, "DavidCombei/wavLM-base-Deepfake_V1"),
|
| 32 |
+
"wavLM-V2": lambda: HuggingFaceFeatureExtractor(WavLMModel, "DavidCombei/wavLM-base-Deepfake_V2"),
|
| 33 |
+
"wavLM-V3": lambda: HuggingFaceFeatureExtractor(WavLMModel, "DavidCombei/wavLM-base-Deepfake_V3"),
|
| 34 |
}
|
| 35 |
|
| 36 |
model1 = joblib.load('model1.joblib')
|