Commit
·
4e3dba8
1
Parent(s):
d26bd2a
fix: NameError: name 'init_empty_weights' is not defined
Browse files- app.py +7 -6
- 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 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 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
|
|
|