Rthur2003 commited on
Commit
8a75b81
·
1 Parent(s): 1d86705

feat: add Wav2Vec2Config and Wav2Vec2MusicClassifier imports for audio classification

Browse files
Files changed (1) hide show
  1. 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):