Update app.py

app.py

@@ -297,47 +297,44 @@ def get_models_data(progress=gr.Progress()):
         """모델 ID를 정규화"""
         return model_id.strip().lower()
         
-    # trending 대신 일반 모델 API 사용
     url = "https://huggingface.co/api/models"
     
     try:
         progress(0, desc="Fetching models data...")
-        params = {
-            'full': 'true',
-            'limit': 1000,
-            'sort': 'lastModified',  # 최신 수정순으로 정렬
-            'direction': -1
-        }
         
-        headers = {'Accept': 'application/json'}
+        # 여러 정렬 방식으로 시도
+        all_found_models = []
+        sort_options = [
+            {'sort': 'downloads', 'direction': -1},
+            {'sort': 'lastModified', 'direction': -1},
+            {'sort': 'likes', 'direction': -1}
+        ]
         
-        response = requests.get(url, params=params, headers=headers)
-        if response.status_code != 200:
-            print(f"API 요청 실패: {response.status_code}")
-            print(f"Response: {response.text}")
-            return create_error_plot(), "<div>모델 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
+        for sort_params in sort_options:
+            params = {
+                'full': 'true',
+                'limit': 1000,
+                **sort_params
+            }
             
-
-        
-        models = response.json()
+            headers = {'Accept': 'application/json'}
+            
+            response = requests.get(url, params=params, headers=headers)
+            if response.status_code == 200:
+                models = response.json()
+                all_found_models.extend(models)
         
-        # target_models와 매칭
+        # 중복 제거
+        seen_ids = set()
         filtered_models = []
-        model_ranks = {}
-        
-        # 전체 순위 정보 저장
-        for idx, model in enumerate(models, 1):
+        for model in all_found_models:
             model_id = normalize_model_id(model.get('id', ''))
-            model_ranks[model_id] = idx
-        
-        # target_models 중 순위권 내 모델 필터링
-        for model in models:
-            if normalize_model_id(model.get('id', '')) in [normalize_model_id(tid) for tid in target_models.keys()]:
-                model['rank'] = model_ranks[normalize_model_id(model.get('id', ''))]
+            if model_id not in seen_ids and model_id in [normalize_model_id(tid) for tid in target_models.keys()]:
+                seen_ids.add(model_id)
                 filtered_models.append(model)
         
-        # 순위로 정렬
-        filtered_models.sort(key=lambda x: x['rank'])
+        # 다운로드 수와 좋아요 수로 정렬
+        filtered_models.sort(key=lambda x: (x.get('downloads', 0), x.get('likes', 0)), reverse=True)
         
         # 순위 할당
         for idx, model in enumerate(filtered_models, 1):
@@ -373,7 +370,7 @@ def get_models_data(progress=gr.Progress()):
         
         fig.update_layout(
             title={
-                'text': 'Hugging Face Models Trending Rankings (Top 1000)',
+                'text': 'Hugging Face Models Rankings by Downloads (Top 1000)',
                 'y':0.95,
                 'x':0.5,
                 'xanchor': 'center',
@@ -397,7 +394,7 @@ def get_models_data(progress=gr.Progress()):
         # HTML 카드 생성
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
-            <h2 style='color: #2c3e50;'>Models Trending Rankings</h2>
+            <h2 style='color: #2c3e50;'>Models Rankings by Downloads</h2>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """