Update app.py

app.py

@@ -293,29 +293,22 @@ target_models = {
 
 def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기"""
-    url = "https://huggingface.co/api/models"  # 기본 API 엔드포인트
-    params = {
-        'full': 'true',
-        'limit': 1000,  # 1000위까지 가져오기
-        'sort': 'trending',  # trending으로 정렬
-        'direction': -1
-    }
+    url = "https://huggingface.co/api/models/trending"  # trending 전용 엔드포인트
     
     try:
         progress(0, desc="Fetching models data...")
-        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}")
-            print(f"Params: {params}")
-            return create_error_plot(), "<div>모델 데이터를 가져오는데 실패했습니다.</div>", pd.DataFrame()
+        params = {
+            'full': 'true',
+            'limit': 1000,  # 1000위까지 가져오기
+            'interval': 'day'  # 일간 트렌딩
+        }
         
+        response = requests.get(url, params=params)
+        response.raise_for_status()
         all_models = response.json()
-        print(f"Total models fetched: {len(all_models)}")  # 디버깅용
+        print(f"Total models fetched: {len(all_models)}")
         
-        # 전체 모델의 순위 정보 저장
+        # 전체 모델의 순위 정보 저장 (1000위까지)
         model_ranks = {model['id']: idx + 1 for idx, model in enumerate(all_models)}
         
         # target_models에 있는 모델만 필터링하고 실제 순위 포함
@@ -324,13 +317,13 @@ def get_models_data(progress=gr.Progress()):
             if model.get('id', '') in target_models:
                 model['rank'] = model_ranks[model['id']]
                 filtered_models.append(model)
-                print(f"Found model: {model['id']} at rank {model['rank']}")  # 디버깅용
+                print(f"Found model: {model['id']} at rank {model['rank']}")
         
         # 순위로 정렬
         filtered_models.sort(key=lambda x: x['rank'])
         
         if not filtered_models:
-            print("No matching models found in the response")  # 디버깅용
+            print("No matching models found in the response")
             return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
         
         progress(0.3, desc="Creating visualization...")
@@ -421,6 +414,8 @@ def get_models_data(progress=gr.Progress()):
                 </a>
             </div>
             """
+            progress((0.6 + 0.3 * filtered_models.index(model)/len(filtered_models)), 
+                    desc=f"Loading model {filtered_models.index(model)+1}/{len(filtered_models)}...")
         
         # 순위권 밖 모델 카드 생성
         for model_id in target_models: