quaja commited on
Commit
1c84924
1 Parent(s): da03f50

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -1,10 +1,10 @@
1
  ---
2
  pipeline_tag: audio-classification
 
 
3
  ---
4
 
5
- from datasets import load_dataset
6
- from transformers import pipeline
7
-
8
-
9
-
10
- classifier = pipeline("audio-classification", model="quaja/hubert-base-amharic-speech-emotion-recognition")
 
1
  ---
2
  pipeline_tag: audio-classification
3
+ metrics:
4
+ - accuracy
5
  ---
6
 
7
+ model_name_or_path = "quaja/hubert-base-amharic-speech-emotion-recognition"
8
+ config = AutoConfig.from_pretrained(model_name_or_path)
9
+ feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
10
+ model = HubertForSpeechClassification.from_pretrained(model_name_or_path).to(device)