iwonachristop commited on
Commit
d00169e
·
1 Parent(s): c7a8332

Update hrefs to models

Browse files
Files changed (1) hide show
  1. src/process_leaderboard_data.py +7 -8
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='models_performance.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,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://github.com/collabora/WhisperSpeech',
46
- 'SpeechT5': 'https://github.com/microsoft/SpeechT5',
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