Update app.py
Browse files
app.py
CHANGED
@@ -37,9 +37,9 @@ class HuggingFaceFeatureExtractor:
|
|
37 |
return outputs.hidden_states[9]
|
38 |
|
39 |
|
40 |
-
truncated_model = CustomWav2Vec2Model.from_pretrained(
|
41 |
FEATURE_EXTRACTOR = HuggingFaceFeatureExtractor(truncated_model, "facebook/wav2vec2-xls-r-2b")
|
42 |
-
classifier, scaler, thresh = joblib.load(
|
43 |
|
44 |
|
45 |
|
|
|
37 |
return outputs.hidden_states[9]
|
38 |
|
39 |
|
40 |
+
truncated_model = CustomWav2Vec2Model.from_pretrained("facebook/wav2vec2-xls-r-2b")
|
41 |
FEATURE_EXTRACTOR = HuggingFaceFeatureExtractor(truncated_model, "facebook/wav2vec2-xls-r-2b")
|
42 |
+
classifier, scaler, thresh = joblib.load('logreg_margin_pruning_ALL_with_scaler_threshold.joblib')
|
43 |
|
44 |
|
45 |
|