Spaces:
Running
Running
feat: add Wav2Vec2Config and Wav2Vec2MusicClassifier imports for audio classification
Browse files- local_demo.py +1 -0
local_demo.py
CHANGED
|
@@ -49,6 +49,7 @@ import numpy as np
|
|
| 49 |
|
| 50 |
from app.training.extract_features_batch import extract_sample_features
|
| 51 |
from app.training.train_deep_classifiers import TorchSklearnWrapper
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
class _AurisUnpickler(pickle.Unpickler):
|
|
|
|
| 49 |
|
| 50 |
from app.training.extract_features_batch import extract_sample_features
|
| 51 |
from app.training.train_deep_classifiers import TorchSklearnWrapper
|
| 52 |
+
from app.training.wav2vec2_classifier import Wav2Vec2Config, Wav2Vec2MusicClassifier
|
| 53 |
|
| 54 |
|
| 55 |
class _AurisUnpickler(pickle.Unpickler):
|