mrfakename commited on
Commit
abcae38
1 Parent(s): 1ec31d9

funcs must be at top

Browse files
Files changed (1) hide show
  1. app.py +48 -42
app.py CHANGED
@@ -19,18 +19,9 @@ DB_PATH = "database.db"
19
  AUDIO_DATASET_ID = "ttseval/tts-arena-new"
20
 
21
  ####################################
22
- # Space initialization
23
  ####################################
24
 
25
- # Download existing DB
26
- print("Downloading DB...")
27
- try:
28
- cache_path = hf_hub_download(repo_id=DB_DATASET_ID, repo_type='dataset', filename=DB_NAME)
29
- shutil.copyfile(cache_path, DB_PATH)
30
- print("Downloaded DB")
31
- except Exception as e:
32
- print("Error while downloading DB:", e)
33
-
34
  def create_db_if_missing():
35
  conn = get_db()
36
  cursor = conn.cursor()
@@ -49,6 +40,53 @@ def create_db_if_missing():
49
  vote INTEGER
50
  );
51
  ''')
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
52
 
53
  # Create DB table (if doesn't exist)
54
  create_db_if_missing()
@@ -191,38 +229,6 @@ model_licenses = {
191
  # return get_random_split(choice)
192
  # else:
193
  # return choice
194
- def get_db():
195
- return sqlite3.connect(DB_PATH)
196
-
197
- def get_leaderboard():
198
- conn = get_db()
199
- cursor = conn.cursor()
200
- cursor.execute('SELECT name, upvote, downvote FROM model WHERE (upvote + downvote) > 5')
201
- data = cursor.fetchall()
202
- df = pd.DataFrame(data, columns=['name', 'upvote', 'downvote'])
203
- df['license'] = df['name'].replace(model_licenses)
204
- df['name'] = df['name'].replace(model_names)
205
- df['votes'] = df['upvote'] + df['downvote']
206
- # df['score'] = round((df['upvote'] / df['votes']) * 100, 2) # Percentage score
207
-
208
- ## ELO SCORE
209
- df['score'] = 1200
210
- for i in range(len(df)):
211
- for j in range(len(df)):
212
- if i != j:
213
- expected_a = 1 / (1 + 10 ** ((df['score'][j] - df['score'][i]) / 400))
214
- expected_b = 1 / (1 + 10 ** ((df['score'][i] - df['score'][j]) / 400))
215
- actual_a = df['upvote'][i] / df['votes'][i]
216
- actual_b = df['upvote'][j] / df['votes'][j]
217
- df.at[i, 'score'] += 32 * (actual_a - expected_a)
218
- df.at[j, 'score'] += 32 * (actual_b - expected_b)
219
- df['score'] = round(df['score'])
220
- ## ELO SCORE
221
- df = df.sort_values(by='score', ascending=False)
222
- df['order'] = ['#' + str(i + 1) for i in range(len(df))]
223
- # df = df[['name', 'score', 'upvote', 'votes']]
224
- df = df[['order', 'name', 'score', 'license', 'votes']]
225
- return df
226
 
227
  # def get_random_splits():
228
  # choice1 = get_random_split()
 
19
  AUDIO_DATASET_ID = "ttseval/tts-arena-new"
20
 
21
  ####################################
22
+ # Functions
23
  ####################################
24
 
 
 
 
 
 
 
 
 
 
25
  def create_db_if_missing():
26
  conn = get_db()
27
  cursor = conn.cursor()
 
40
  vote INTEGER
41
  );
42
  ''')
43
+ def get_db():
44
+ return sqlite3.connect(DB_PATH)
45
+
46
+ def get_leaderboard():
47
+ conn = get_db()
48
+ cursor = conn.cursor()
49
+ cursor.execute('SELECT name, upvote, downvote FROM model WHERE (upvote + downvote) > 5')
50
+ data = cursor.fetchall()
51
+ df = pd.DataFrame(data, columns=['name', 'upvote', 'downvote'])
52
+ df['license'] = df['name'].replace(model_licenses)
53
+ df['name'] = df['name'].replace(model_names)
54
+ df['votes'] = df['upvote'] + df['downvote']
55
+ # df['score'] = round((df['upvote'] / df['votes']) * 100, 2) # Percentage score
56
+
57
+ ## ELO SCORE
58
+ df['score'] = 1200
59
+ for i in range(len(df)):
60
+ for j in range(len(df)):
61
+ if i != j:
62
+ expected_a = 1 / (1 + 10 ** ((df['score'][j] - df['score'][i]) / 400))
63
+ expected_b = 1 / (1 + 10 ** ((df['score'][i] - df['score'][j]) / 400))
64
+ actual_a = df['upvote'][i] / df['votes'][i]
65
+ actual_b = df['upvote'][j] / df['votes'][j]
66
+ df.at[i, 'score'] += 32 * (actual_a - expected_a)
67
+ df.at[j, 'score'] += 32 * (actual_b - expected_b)
68
+ df['score'] = round(df['score'])
69
+ ## ELO SCORE
70
+ df = df.sort_values(by='score', ascending=False)
71
+ df['order'] = ['#' + str(i + 1) for i in range(len(df))]
72
+ # df = df[['name', 'score', 'upvote', 'votes']]
73
+ df = df[['order', 'name', 'score', 'license', 'votes']]
74
+ return df
75
+
76
+
77
+ ####################################
78
+ # Space initialization
79
+ ####################################
80
+
81
+ # Download existing DB
82
+ print("Downloading DB...")
83
+ try:
84
+ cache_path = hf_hub_download(repo_id=DB_DATASET_ID, repo_type='dataset', filename=DB_NAME)
85
+ shutil.copyfile(cache_path, DB_PATH)
86
+ print("Downloaded DB")
87
+ except Exception as e:
88
+ print("Error while downloading DB:", e)
89
+
90
 
91
  # Create DB table (if doesn't exist)
92
  create_db_if_missing()
 
229
  # return get_random_split(choice)
230
  # else:
231
  # return choice
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
232
 
233
  # def get_random_splits():
234
  # choice1 = get_random_split()