From 7fd191edd9a505af312467d6f00fede29cff0da1 Mon Sep 17 00:00:00 2001 From: harshit katyal Date: Sun, 12 Dec 2021 20:53:33 +0000 Subject: [PATCH] Update README.md --- README.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 249c71b..0b9128a 100644 --- a/README.md +++ b/README.md @@ -30,7 +30,7 @@ import pandas as pd ~~~ ~~~ device = torch.device("cuda" if torch.cuda.is_available() else "cpu") -model_name_or_path = "Bagus/wav2vec2-xlsr-greek-speech-emotion-recognition" +model_name_or_path = "harshit345/xlsr-wav2vec-speech-emotion-recognition" config = AutoConfig.from_pretrained(model_name_or_path) feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(model_name_or_path) sampling_rate = feature_extractor.sampling_rate @@ -59,10 +59,10 @@ path = '/data/jtes_v1.1/wav/f01/ang/f01_ang_01.wav' outputs = predict(path, sampling_rate) ~~~ ~~~ -[{'Emotion': 'anger', 'Score': '98.3%'}, - {'Emotion': 'disgust', 'Score': '0.0%'}, - {'Emotion': 'fear', 'Score': '0.4%'}, - {'Emotion': 'happiness', 'Score': '0.7%'}, +[{'Emotion': 'anger', 'Score': '78.3%'}, + {'Emotion': 'disgust', 'Score': '11.7%'}, + {'Emotion': 'fear', 'Score': '5.4%'}, + {'Emotion': 'happiness', 'Score': '4.1%'}, {'Emotion': 'sadness', 'Score': '0.5%'}] ~~~