openfree commited on
Commit
4991d37
ยท
verified ยท
1 Parent(s): 0a5f955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -7
app.py CHANGED
@@ -304,8 +304,8 @@ def get_models_data(progress=gr.Progress()):
304
  # ์ „์ฒด ๋ชจ๋ธ ๋ชฉ๋ก ๊ฐ€์ ธ์˜ค๊ธฐ (์ข‹์•„์š” ์ˆœ์œผ๋กœ ์ •๋ ฌ)
305
  global_params = {
306
  'full': 'true',
307
- 'limit': 3000,
308
- 'sort': 'likes', # ์ข‹์•„์š” ์ˆœ์œผ๋กœ ๋ณ€๊ฒฝ
309
  'direction': -1
310
  }
311
 
@@ -349,7 +349,7 @@ def get_models_data(progress=gr.Progress()):
349
  rank_info = global_ranks.get(normalized_id, {})
350
  model_info = {
351
  'id': model_id,
352
- 'global_rank': rank_info.get('rank', 'Not in top 3000'),
353
  'downloads': model_data.get('downloads', 0),
354
  'likes': model_data.get('likes', 0),
355
  'title': model_data.get('title', 'No Title')
@@ -360,7 +360,7 @@ def get_models_data(progress=gr.Progress()):
360
  print(f"Failed to fetch data for {model_id}: {response.status_code}")
361
  filtered_models.append({
362
  'id': model_id,
363
- 'global_rank': 'Not in top 3000',
364
  'downloads': 0,
365
  'likes': 0,
366
  'title': 'No Title'
@@ -369,7 +369,7 @@ def get_models_data(progress=gr.Progress()):
369
  print(f"Error fetching data for {model_id}: {str(e)}")
370
  filtered_models.append({
371
  'id': model_id,
372
- 'global_rank': 'Not in top 3000',
373
  'downloads': 0,
374
  'likes': 0,
375
  'title': 'No Title'
@@ -401,10 +401,13 @@ def get_models_data(progress=gr.Progress()):
401
  valid_likes = [likes[i] for i in valid_indices]
402
  valid_downloads = [downloads[i] for i in valid_indices]
403
 
 
 
 
404
  # ๋ง‰๋Œ€ ๊ทธ๋ž˜ํ”„ ์ƒ์„ฑ
405
  fig.add_trace(go.Bar(
406
  x=valid_ids,
407
- y=valid_ranks,
408
  text=[f"Global Rank: {r}<br>Likes: {format(l, ',')}<br>Downloads: {format(d, ',')}"
409
  for r, l, d in zip(valid_ranks, valid_likes, valid_downloads)],
410
  textposition='auto',
@@ -422,7 +425,11 @@ def get_models_data(progress=gr.Progress()):
422
  },
423
  xaxis_title='Model ID',
424
  yaxis_title='Global Rank',
425
- yaxis_autorange='reversed', # ์ˆœ์œ„๋ฅผ ์œ„์—์„œ ์•„๋ž˜๋กœ
 
 
 
 
426
  height=800,
427
  showlegend=False,
428
  template='plotly_white',
 
304
  # ์ „์ฒด ๋ชจ๋ธ ๋ชฉ๋ก ๊ฐ€์ ธ์˜ค๊ธฐ (์ข‹์•„์š” ์ˆœ์œผ๋กœ ์ •๋ ฌ)
305
  global_params = {
306
  'full': 'true',
307
+ 'limit': 10000, # 10000์œ„๊นŒ์ง€ ํ™•์žฅ
308
+ 'sort': 'likes',
309
  'direction': -1
310
  }
311
 
 
349
  rank_info = global_ranks.get(normalized_id, {})
350
  model_info = {
351
  'id': model_id,
352
+ 'global_rank': rank_info.get('rank', 'Not in top 10000'),
353
  'downloads': model_data.get('downloads', 0),
354
  'likes': model_data.get('likes', 0),
355
  'title': model_data.get('title', 'No Title')
 
360
  print(f"Failed to fetch data for {model_id}: {response.status_code}")
361
  filtered_models.append({
362
  'id': model_id,
363
+ 'global_rank': 'Not in top 10000',
364
  'downloads': 0,
365
  'likes': 0,
366
  'title': 'No Title'
 
369
  print(f"Error fetching data for {model_id}: {str(e)}")
370
  filtered_models.append({
371
  'id': model_id,
372
+ 'global_rank': 'Not in top 10000',
373
  'downloads': 0,
374
  'likes': 0,
375
  'title': 'No Title'
 
401
  valid_likes = [likes[i] for i in valid_indices]
402
  valid_downloads = [downloads[i] for i in valid_indices]
403
 
404
+ # Y์ถ• ๊ฐ’์„ ๋ฐ˜์ „
405
+ y_values = [10001 - r if isinstance(r, (int, float)) else 0 for r in valid_ranks]
406
+
407
  # ๋ง‰๋Œ€ ๊ทธ๋ž˜ํ”„ ์ƒ์„ฑ
408
  fig.add_trace(go.Bar(
409
  x=valid_ids,
410
+ y=y_values,
411
  text=[f"Global Rank: {r}<br>Likes: {format(l, ',')}<br>Downloads: {format(d, ',')}"
412
  for r, l, d in zip(valid_ranks, valid_likes, valid_downloads)],
413
  textposition='auto',
 
425
  },
426
  xaxis_title='Model ID',
427
  yaxis_title='Global Rank',
428
+ yaxis=dict(
429
+ ticktext=[str(i) for i in range(1, 10001, 500)],
430
+ tickvals=[10001 - i for i in range(1, 10001, 500)],
431
+ range=[0, 10000]
432
+ ),
433
  height=800,
434
  showlegend=False,
435
  template='plotly_white',