greenarcade commited on
Commit
4e3dba8
·
1 Parent(s): d26bd2a

fix: NameError: name 'init_empty_weights' is not defined

Browse files
Files changed (2) hide show
  1. app.py +7 -6
  2. requirements.txt +0 -0
app.py CHANGED
@@ -4,12 +4,13 @@ import librosa
4
  from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2FeatureExtractor
5
 
6
  model_name = "greenarcade/wav2vec2-vd-bird-sound-classification"
7
- model = Wav2Vec2ForSequenceClassification.from_pretrained(
8
- model_name,
9
- local_files_only=False,
10
- use_auth_token=None,
11
- trust_remote_code=False
12
- )
 
13
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name)
14
 
15
  def predict(audio_file):
 
4
  from transformers import Wav2Vec2ForSequenceClassification, Wav2Vec2FeatureExtractor
5
 
6
  model_name = "greenarcade/wav2vec2-vd-bird-sound-classification"
7
+ # model = Wav2Vec2ForSequenceClassification.from_pretrained(
8
+ # model_name,
9
+ # local_files_only=False,
10
+ # use_auth_token=None,
11
+ # trust_remote_code=False
12
+ # )
13
+ model = Wav2Vec2ForSequenceClassification.from_pretrained(model_name)
14
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name)
15
 
16
  def predict(audio_file):
requirements.txt CHANGED
Binary files a/requirements.txt and b/requirements.txt differ