Update app.py

app.py

@@ -318,53 +318,76 @@ def get_korea_models():
         return []
 
 def get_all_models(limit=1000):
-    """페이지네이션을 사용하여 모든 모델 가져오기"""
+    """모든 모델과 Korea 관련 모델 가져오기"""
     all_models = []
-    offset = 0
-    page_size = 1000  # API의 기본 페이지 크기
-
-    # 일반 모델 가져오기
-    while len(all_models) < limit:
-        params = {
-            "limit": min(page_size, limit - len(all_models)),
-            "full": "True",
-            "config": "True",
-            "offset": offset
-        }
-        
-        response = requests.get(
-            "https://huggingface.co/api/models",
-            headers={'Accept': 'application/json'},
-            params=params
-        )
-        
-        if response.status_code != 200:
-            break
-            
-        models = response.json()
-        if not models:  # 더 이상 데이터가 없으면 종료
-            break
-            
-        all_models.extend(models)
-        offset += len(models)
+    
+    # 1. 일반 모델 리스트 가져오기
+    params = {
+        "limit": limit,
+        "full": "True",
+        "config": "True"
+    }
+    
+    response = requests.get(
+        "https://huggingface.co/api/models",
+        headers={'Accept': 'application/json'},
+        params=params
+    )
+    
+    if response.status_code == 200:
+        all_models.extend(response.json())
+        print(f"Fetched {len(all_models)} general models")
+    
+    # 2. Korea 검색 결과 가져오기
+    korea_params = {
+        "search": "korea",
+        "full": "True",
+        "config": "True",
+        "limit": limit
+    }
+    
+    korea_response = requests.get(
+        "https://huggingface.co/api/models",
+        headers={'Accept': 'application/json'},
+        params=korea_params
+    )
+    
+    if korea_response.status_code == 200:
+        korea_models = korea_response.json()
+        print(f"Fetched {len(korea_models)} Korea-related models")
         
-        if len(models) < page_size:  # 마지막 페이지면 종료
-            break
+        # 중복 제거하면서 Korea 모델 추가
+        existing_ids = {model.get('id', '') for model in all_models}
+        for model in korea_models:
+            if model.get('id', '') not in existing_ids:
+                all_models.append(model)
+                existing_ids.add(model.get('id', ''))
     
-    # Korea 관련 모델 추가
-    korea_models = get_korea_models()
-    print(f"Found {len(korea_models)} Korea-related models")
+    # 3. Korean 검색 결과 가져오기
+    korean_params = {
+        "search": "korean",
+        "full": "True",
+        "config": "True",
+        "limit": limit
+    }
     
-    # 중복 제거를 위한 ID 세트
-    existing_ids = {model.get('id', '') for model in all_models}
+    korean_response = requests.get(
+        "https://huggingface.co/api/models",
+        headers={'Accept': 'application/json'},
+        params=korean_params
+    )
     
-    # 중복되지 않은 Korea 모델만 추가
-    for model in korea_models:
-        if model.get('id', '') not in existing_ids:
-            all_models.append(model)
-            existing_ids.add(model.get('id', ''))
+    if korean_response.status_code == 200:
+        korean_models = korean_response.json()
+        print(f"Fetched {len(korean_models)} Korean-related models")
+        
+        # 중복 제거하면서 Korean 모델 추가
+        for model in korean_models:
+            if model.get('id', '') not in existing_ids:
+                all_models.append(model)
+                existing_ids.add(model.get('id', ''))
     
-    print(f"Total models after adding Korea-related: {len(all_models)}")
+    print(f"Total unique models: {len(all_models)}")
     return all_models[:limit]
 
 def get_models_data(progress=gr.Progress()):
@@ -372,7 +395,7 @@ def get_models_data(progress=gr.Progress()):
     try:
         progress(0, desc="Fetching models...")
         
-        # 페이지네이션을 사용하여 모델 가져오기
+        # 모델 가져오기
         all_global_models = get_all_models(limit=1000)
         print(f"Actually fetched models count: {len(all_global_models)}")