Update app.py

app.py

@@ -291,67 +291,71 @@ target_models = {
     "sel303/llama3-diverce-ver1.6": "https://huggingface.co/sel303/llama3-diverce-ver1.6"    
 }
 
+# all_models 관련 코드 제거하고 get_models_data 함수 내부에서 처리하도록 수정
+
 def get_models_data(progress=gr.Progress()):
     """모델 데이터 가져오기"""
-    url = "https://huggingface.co/api/models/trending"  # trending API 사용
+    url = "https://huggingface.co/api/models"  # 기본 모델 API 엔드포인트
     
     try:
         progress(0, desc="Fetching models data...")
-        params = {
-            'full': 'true',
-            'limit': 1000
-        }
         
-        headers = {
-            'Accept': 'application/json'
-        }
-        
-        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()
-        
-        models = response.json()
-        print(f"Total models fetched: {len(models)}")
-        
-        # target_models와 매칭
-        filtered_models = []
-        model_ranks = {}
+        # 여러 정렬 방식으로 시도
+        all_found_models = []
+        sort_options = [
+            {'sort': 'downloads', 'direction': -1},
+            {'sort': 'lastModified', 'direction': -1},
+            {'sort': 'likes', 'direction': -1}
+        ]
         
-        # 전체 순위 정보 저장
-        for idx, model in enumerate(models, 1):
+        for sort_params in sort_options:
+            params = {
+                'full': 'true',
+                'limit': 1000,
+                **sort_params
+            }
+            
+            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)
+        
+        # 중복 제거
+        seen_ids = set()
+        unique_models = []
+        for model in all_found_models:
             model_id = model.get('id', '')
-            model_ranks[model_id] = idx
-        
-        # target_models 중 순위권 내 모델 필터링
-        for model in models:
-            if model.get('id', '') in target_models:
-                model['rank'] = model_ranks[model.get('id', '')]
-                filtered_models.append(model)
-                print(f"Found model: {model.get('id', '')} at rank {model['rank']}")
+            if model_id not in seen_ids and model_id in target_models:
+                seen_ids.add(model_id)
+                unique_models.append(model)
         
-        # 순위로 정렬
-        filtered_models.sort(key=lambda x: x['rank'])
+        # 다운로드 수로 정렬
+        unique_models.sort(key=lambda x: x.get('downloads', 0), reverse=True)
         
-        print(f"Matched {len(filtered_models)} models out of {len(target_models)} targets")
+        # 순위 할당
+        for idx, model in enumerate(unique_models, 1):
+            model['rank'] = idx
         
-        if not filtered_models:
+        if not unique_models:
             return create_error_plot(), "<div>선택된 모델의 데이터를 찾을 수 없습니다.</div>", pd.DataFrame()
         
+        # 나머지 시각화 코드는 동일하게 유지
         progress(0.3, desc="Creating visualization...")
         
         # 시각화 생성
         fig = go.Figure()
         
         # 데이터 준비
-        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]
-        downloads = [model.get('downloads', 0) for model in filtered_models]
+        ids = [model['id'] for model in unique_models]
+        ranks = [model['rank'] for model in unique_models]
+        likes = [model.get('likes', 0) for model in unique_models]
+        downloads = [model.get('downloads', 0) for model in unique_models]
         
-        # Y축 값을 반전 (1000 - rank + 1)
+        # Y축 값을 반전
         y_values = [1001 - r for r in ranks]
+
         
         # 막대 그래프 생성
         fig.add_trace(go.Bar(