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 |
|