Commit
·
d00169e
1
Parent(s):
c7a8332
Update hrefs to models
Browse files
src/process_leaderboard_data.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
|
4 |
|
5 |
class Leaderboard():
|
6 |
-
def __init__(self, data_path='
|
7 |
self.data_path = data_path
|
8 |
self.models_performance_dataframe = self.initialize_models_performance_dataframe()
|
9 |
self.link_dict = self.initialize_link_dict()
|
@@ -11,7 +11,6 @@ class Leaderboard():
|
|
11 |
self.emotional_dataset_list = sorted(self.initialize_emotional_datasets())
|
12 |
self.emotion_list = sorted(list(set(self.models_performance_dataframe['emotion'])))
|
13 |
|
14 |
-
|
15 |
def get_dataset_list(self) -> list:
|
16 |
return self.dataset_list
|
17 |
|
@@ -42,11 +41,11 @@ class Leaderboard():
|
|
42 |
# Dictionary mapping model names to their GitHub repository links
|
43 |
def initialize_link_dict(self) -> dict:
|
44 |
return {
|
45 |
-
'WhisperSpeech': 'https://
|
46 |
-
'
|
47 |
-
'VALL-E-X': 'https://github.com/Plachtaa/VALL-E-X',
|
48 |
-
'XTTS-v2': 'https://huggingface.co/coqui/XTTS-v2',
|
49 |
-
'OuteTTS': 'https://huggingface.co/OuteAI/OuteTTS-0.2-500M'
|
50 |
}
|
51 |
|
52 |
|
@@ -102,7 +101,7 @@ class Leaderboard():
|
|
102 |
# Map model names to hyperlinks using the link_dict.
|
103 |
leaderboard_dataframe['Model'] = leaderboard_dataframe['Model'].map(lambda model: f"[{model}]({self.link_dict.get(model, '')})")
|
104 |
leaderboard_dataframe.sort_values(by='Average', ascending=False, inplace=True)
|
105 |
-
|
106 |
return leaderboard_dataframe
|
107 |
|
108 |
|
|
|
3 |
|
4 |
|
5 |
class Leaderboard():
|
6 |
+
def __init__(self, data_path='results.csv'):
|
7 |
self.data_path = data_path
|
8 |
self.models_performance_dataframe = self.initialize_models_performance_dataframe()
|
9 |
self.link_dict = self.initialize_link_dict()
|
|
|
11 |
self.emotional_dataset_list = sorted(self.initialize_emotional_datasets())
|
12 |
self.emotion_list = sorted(list(set(self.models_performance_dataframe['emotion'])))
|
13 |
|
|
|
14 |
def get_dataset_list(self) -> list:
|
15 |
return self.dataset_list
|
16 |
|
|
|
41 |
# Dictionary mapping model names to their GitHub repository links
|
42 |
def initialize_link_dict(self) -> dict:
|
43 |
return {
|
44 |
+
'WhisperSpeech/WhisperSpeech': 'https://huggingface.co/WhisperSpeech/WhisperSpeech',
|
45 |
+
'microsoft/speecht5_vc': 'https://huggingface.co/microsoft/speecht5_vc',
|
46 |
+
'Plachtaa/VALL-E-X': 'https://github.com/Plachtaa/VALL-E-X',
|
47 |
+
'coqui/XTTS-v2': 'https://huggingface.co/coqui/XTTS-v2',
|
48 |
+
'OuteAI/OuteTTS-0.2-500M': 'https://huggingface.co/OuteAI/OuteTTS-0.2-500M'
|
49 |
}
|
50 |
|
51 |
|
|
|
101 |
# Map model names to hyperlinks using the link_dict.
|
102 |
leaderboard_dataframe['Model'] = leaderboard_dataframe['Model'].map(lambda model: f"[{model}]({self.link_dict.get(model, '')})")
|
103 |
leaderboard_dataframe.sort_values(by='Average', ascending=False, inplace=True)
|
104 |
+
leaderboard_dataframe.insert(0, "Rank", range(1, len(leaderboard_dataframe) + 1))
|
105 |
return leaderboard_dataframe
|
106 |
|
107 |
|