Update app.py

app.py

@@ -292,23 +292,19 @@ target_models = {
     "sel303/llama3-diverce-ver1.6": "https://huggingface.co/sel303/llama3-diverce-ver1.6"    
 }
 
-
-
 def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기"""
     def normalize_model_id(model_id):
         """모델 ID를 정규화"""
         return model_id.strip().lower()
         
-    url = "https://huggingface.co/api/models"
+    url = "https://huggingface.co/api/models/trending"  # trending API로 변경
     
     try:
         progress(0, desc="Fetching models data...")
         params = {
             'full': 'true',
-            'limit': 3000,  # 3000개로 증가
-            'sort': 'likes',
-            'direction': -1
+            'limit': 3000
         }
         
         headers = {'Accept': 'application/json'}
@@ -321,7 +317,7 @@ def get_models_data(progress=gr.Progress()):
         
         models = response.json()
         
-        # 전체 순위 정보 저장 (다운로드 수 기준)
+        # 전체 순위 정보 저장 (trending 순위 기준)
         model_ranks = {}
         model_data = {}  # 모든 모델의 상세 데이터 저장
         
@@ -338,8 +334,6 @@ def get_models_data(progress=gr.Progress()):
         filtered_models = []
         for target_id in target_models.keys():
             normalized_target_id = normalize_model_id(target_id)
-            
-            # 먼저 전체 순위에서 찾기
             if normalized_target_id in model_data:
                 model_info = {
                     'id': target_id,
@@ -349,34 +343,18 @@ def get_models_data(progress=gr.Progress()):
                     'title': model_data[normalized_target_id]['title']
                 }
             else:
-                # 순위권 밖의 모델은 개별 API 호출로 정보 가져오기
-                try:
-                    model_url = f"https://huggingface.co/api/models/{target_id}"
-                    model_response = requests.get(model_url, headers=headers)
-                    if model_response.status_code == 200:
-                        model_info = model_response.json()
-                        model_info['id'] = target_id
-                        model_info['rank'] = 'Not in top 3000'
-                    else:
-                        model_info = {
-                            'id': target_id,
-                            'rank': 'Not in top 3000',
-                            'downloads': 0,
-                            'likes': 0,
-                            'title': 'No Title'
-                        }
-                except Exception as e:
-                    print(f"Error fetching data for model {target_id}: {str(e)}")
-                    model_info = {
-                        'id': target_id,
-                        'rank': 'Not in top 3000',
-                        'downloads': 0,
-                        'likes': 0,
-                        'title': 'No Title'
-                    }
-            
+                model_info = {
+                    'id': target_id,
+                    'rank': 'Not in trending',
+                    'downloads': 'N/A',
+                    'likes': 'N/A',
+                    'title': 'No Title'
+                }
             filtered_models.append(model_info)
         
+        # trending 순위로 정렬
+        filtered_models.sort(key=lambda x: float('inf') if x['rank'] == 'Not in trending' else x['rank'])
+        
         # 순위로 정렬 (순위가 숫자인 경우만)
         filtered_models.sort(key=lambda x: (
             float('inf') if x['rank'] == 'Not in top 3000' else x['rank']