Update app.py

app.py

@@ -18,71 +18,53 @@ target_models = {
 }
 
 def get_models_data(progress=gr.Progress()):
-    """모델 데이터 가져오기 (스페이스와 동일한 방식)"""
-    url = "https://huggingface.co/api/models"  # 기본 API 엔드포인트
+    """모델 데이터 가져오기"""
+    url = "https://huggingface.co/api/models"
     params = {
         'full': 'true',
-        'limit': 1000,
-        'sort': 'lastModified',  # 최신 순으로 정렬
-        'direction': -1
+        'limit': 300  # 스페이스와 동일하게 300개로 설정
     }
     
-    headers = {
-        'Accept': 'application/json',
-        'User-Agent': 'Mozilla/5.0'
-    }
-    
-    # API 토큰이 있는 경우 헤더에 추가
-    if HF_TOKEN:
-        headers['Authorization'] = f'Bearer {HF_TOKEN}'
-    
     try:
         progress(0, desc="Fetching models data...")
-        response = requests.get(url, params=params, headers=headers)
+        response = requests.get(url, params=params)
         
         if response.status_code != 200:
             print(f"API 요청 실패: {response.status_code}")
             print(f"Response: {response.text}")
+            print(f"URL: {url}")
             return create_error_plot(), "<div>모델 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
         
         all_models = response.json()
         
-        # 순위 정보 저장
-        model_ranks = {model['id']: idx + 1 for idx, model in enumerate(all_models)}
-        
-        # target_models 필터링 및 순위 정보 포함
-        models = []
+        # target_models에 있는 모델만 필터링
+        filtered_models = []
         for model in all_models:
             if model.get('id', '') in target_models:
-                model['rank'] = model_ranks.get(model['id'], 'N/A')
-                models.append(model)
+                model['rank'] = len(filtered_models) + 1
+                filtered_models.append(model)
         
-        if not models:
+        if not filtered_models:
             return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
         
-        # 순위별로 정렬
-        models.sort(key=lambda x: x['rank'])
-        
         progress(0.3, desc="Creating visualization...")
         
         # 시각화 생성
         fig = go.Figure()
         
         # 데이터 준비
-        ids = [model['id'] for model in models]
-        ranks = [model['rank'] for model in models]
-        likes = [model.get('likes', 0) for model in models]
-        downloads = [model.get('downloads', 0) for model in models]
+        ids = [model['id'] for model in filtered_models]
+        ranks = [model['rank'] for model in filtered_models]
+        likes = [model.get('likes', 0) for model in filtered_models]
         
-        # Y축 값을 반전 (1000 - rank + 1)
-        y_values = [1001 - r for r in ranks]
+        # Y축 값을 반전 (300 - rank + 1)
+        y_values = [301 - r for r in ranks]
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(
             x=ids,
             y=y_values,
-            text=[f"Rank: {r}<br>Likes: {l}<br>Downloads: {d}" 
-                  for r, l, d in zip(ranks, likes, downloads)],
+            text=[f"Rank: {r}<br>Likes: {l}" for r, l in zip(ranks, likes)],
             textposition='auto',
             marker_color='rgb(158,202,225)',
             opacity=0.8
@@ -90,7 +72,7 @@ def get_models_data(progress=gr.Progress()):
         
         fig.update_layout(
             title={
-                'text': 'Hugging Face Models Rankings',
+                'text': 'Hugging Face Models Rankings (Top 300)',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -99,9 +81,9 @@ def get_models_data(progress=gr.Progress()):
             xaxis_title='Model ID',
             yaxis_title='Rank',
             yaxis=dict(
-                ticktext=[str(i) for i in range(1, 1001, 50)],
-                tickvals=[1001 - i for i in range(1, 1001, 50)],
-                range=[0, 1000]
+                ticktext=[str(i) for i in range(1, 301, 20)],
+                tickvals=[301 - i for i in range(1, 301, 20)],
+                range=[0, 300]
             ),
             height=800,
             showlegend=False,
@@ -118,12 +100,11 @@ def get_models_data(progress=gr.Progress()):
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
-        for model in models:
+        for model in filtered_models:
             model_id = model.get('id', '')
             rank = model.get('rank', 'N/A')
             likes = model.get('likes', 0)
             downloads = model.get('downloads', 0)
-            description = model.get('description', 'No description available')[:200]
             
             html_content += f"""
             <div style='
@@ -136,7 +117,6 @@ def get_models_data(progress=gr.Progress()):
                 <h3 style='color: #34495e;'>Rank #{rank} - {model_id}</h3>
                 <p style='color: #7f8c8d;'>👍 Likes: {likes}</p>
                 <p style='color: #7f8c8d;'>⬇️ Downloads: {downloads}</p>
-                <p style='color: #7f8c8d; font-size: 0.9em;'>{description}...</p>
                 <a href='{target_models[model_id]}' 
                    target='_blank' 
                    style='
@@ -162,7 +142,7 @@ def get_models_data(progress=gr.Progress()):
             'Likes': model.get('likes', 'N/A'),
             'Downloads': model.get('downloads', 'N/A'),
             'URL': target_models[model.get('id', '')]
-        } for model in models])
+        } for model in filtered_models])
         
         progress(1.0, desc="Complete!")
         return fig, html_content, df
@@ -555,8 +535,8 @@ def refresh_data():
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("""
-    # 🤗 HuggingFace Rankings Analysis
-    실시간으로 Hugging Face의 Spaces와 Models 순위를 분석합니다.
+    # 🤗 허깅페이스 '한국 리더보드'
+    실시간으로 Hugging Face의 Spaces와 Models 인기 순위를 분석합니다. 신규 등록 요청: [email protected]
     """)
     
     with gr.Tab("Spaces Trending"):