harshit345 commited on
Commit
7fd191e
1 Parent(s): 696c231

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -30,7 +30,7 @@ import pandas as pd
30
  ~~~
31
  ~~~
32
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
33
- model_name_or_path = "Bagus/wav2vec2-xlsr-greek-speech-emotion-recognition"
34
  config = AutoConfig.from_pretrained(model_name_or_path)
35
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
36
  sampling_rate = feature_extractor.sampling_rate
@@ -59,10 +59,10 @@ path = '/data/jtes_v1.1/wav/f01/ang/f01_ang_01.wav'
59
  outputs = predict(path, sampling_rate)
60
  ~~~
61
  ~~~
62
- [{'Emotion': 'anger', 'Score': '98.3%'},
63
- {'Emotion': 'disgust', 'Score': '0.0%'},
64
- {'Emotion': 'fear', 'Score': '0.4%'},
65
- {'Emotion': 'happiness', 'Score': '0.7%'},
66
  {'Emotion': 'sadness', 'Score': '0.5%'}]
67
  ~~~
68
 
 
30
  ~~~
31
  ~~~
32
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
33
+ model_name_or_path = "harshit345/xlsr-wav2vec-speech-emotion-recognition"
34
  config = AutoConfig.from_pretrained(model_name_or_path)
35
  feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path)
36
  sampling_rate = feature_extractor.sampling_rate
 
59
  outputs = predict(path, sampling_rate)
60
  ~~~
61
  ~~~
62
+ [{'Emotion': 'anger', 'Score': '78.3%'},
63
+ {'Emotion': 'disgust', 'Score': '11.7%'},
64
+ {'Emotion': 'fear', 'Score': '5.4%'},
65
+ {'Emotion': 'happiness', 'Score': '4.1%'},
66
  {'Emotion': 'sadness', 'Score': '0.5%'}]
67
  ~~~
68