Update app.py

app.py

@@ -437,22 +437,20 @@ def get_trending_spaces(progress=gr.Progress()) -> Tuple[str, str]:
         progress(0, desc="Fetching spaces data...")
         params = {
             'full': 'true',
-            'limit': 10
+            'limit': 10,  # 상위 10개만 가져오도록 수정
+            'sort': 'likes'  # 좋아요 순으로 정렬
         }
         response = requests.get(url, params=params)
         response.raise_for_status()
         spaces = response.json()
         
-        # 상위 10개만 선택
-        top_spaces = spaces[:10]
-        
         progress(0.1, desc="Creating gallery...")
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
-        for idx, space in enumerate(top_spaces):
+        for idx, space in enumerate(spaces):
             html_content += get_card(space, idx, "space")
             progress((0.1 + 0.9 * idx/10), desc=f"Loading space {idx+1}/10...")
             
@@ -471,34 +469,23 @@ def get_models(progress=gr.Progress()) -> Tuple[str, str]:
     
     try:
         progress(0, desc="Fetching models data...")
-        response = requests.get(url)
+        params = {
+            'full': 'true',
+            'limit': 10,  # 상위 10개만 가져오도록 수정
+            'sort': 'likes'  # 좋아요 순으로 정렬
+        }
+        response = requests.get(url, params=params)
         response.raise_for_status()
         models = response.json()
         
-        # 상위 10개만 선택
-        top_models = models[:10]
-        
         progress(0.1, desc="Creating gallery...")
-        
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
-        for idx, model in enumerate(top_models):
-            model_id = model.get('id', '')
-            author = model_id.split('/')[0]
-            title = model_id.split('/')[-1]
-            likes = format(model.get('likes', 0), ',')
-            downloads = format(model.get('downloads', 0), ',')
-            created = model.get('createdAt', '').split('T')[0]
-            url = f"https://huggingface.co/{model_id}"
-            
-            screenshot = get_cached_screenshot(url)
+        for idx, model in enumerate(models):
             html_content += get_card(model, idx, "model")
-            
-
-            
             progress((0.1 + 0.9 * idx/10), desc=f"Loading model {idx+1}/10...")
             
         html_content += "</div></div>"
@@ -516,33 +503,23 @@ def get_datasets(progress=gr.Progress()) -> Tuple[str, str]:
     
     try:
         progress(0, desc="Fetching datasets data...")
-        response = requests.get(url)
+        params = {
+            'full': 'true',
+            'limit': 10,  # 상위 10개만 가져오도록 수정
+            'sort': 'likes'  # 좋아요 순으로 정렬
+        }
+        response = requests.get(url, params=params)
         response.raise_for_status()
         datasets = response.json()
         
-        # 상위 10개만 선택
-        top_datasets = datasets[:10]
-        
         progress(0.1, desc="Creating gallery...")
         html_content = """
         <div style='padding: 20px; background: #f5f5f5;'>
             <div style='display: grid; grid-template-columns: repeat(auto-fill, minmax(300px, 1fr)); gap: 20px;'>
         """
         
-        for idx, dataset in enumerate(top_datasets):
-            dataset_id = dataset.get('id', '')
-            author = dataset_id.split('/')[0]
-            title = dataset_id.split('/')[-1]
-            likes = format(dataset.get('likes', 0), ',')
-            downloads = format(dataset.get('downloads', 0), ',')
-            created = dataset.get('createdAt', '').split('T')[0]
-            url = f"https://huggingface.co/datasets/{dataset_id}"
-            
-            screenshot = get_cached_screenshot(url)
+        for idx, dataset in enumerate(datasets):
             html_content += get_card(dataset, idx, "dataset")
-            
-
-            
             progress((0.1 + 0.9 * idx/10), desc=f"Loading dataset {idx+1}/10...")
             
         html_content += "</div></div>"
@@ -562,7 +539,7 @@ def create_interface():
         with gr.Tabs() as tabs:
             # Spaces 탭
             with gr.Tab("🎯 Trending Spaces"):
-                gr.Markdown("Shows top 300 trending spaces on Hugging Face")
+                gr.Markdown("Shows top 10 trending spaces with AI ratings")
                 with gr.Row():
                     spaces_refresh_btn = gr.Button("Refresh Spaces", variant="primary")
                 spaces_gallery = gr.HTML()
@@ -570,7 +547,7 @@ def create_interface():
             
             # Models 탭
             with gr.Tab("🤖 Trending Models"):
-                gr.Markdown("Shows top 10 trending models on Hugging Face")
+                gr.Markdown("Shows top 10 trending models with AI ratings")
                 with gr.Row():
                     models_refresh_btn = gr.Button("Refresh Models", variant="primary")
                 models_gallery = gr.HTML()
@@ -578,7 +555,7 @@ def create_interface():
             
             # Datasets 탭
             with gr.Tab("📊 Trending Datasets"):
-                gr.Markdown("Shows top 10 trending datasets on Hugging Face")
+                gr.Markdown("Shows top 10 trending datasets with AI ratings")
                 with gr.Row():
                     datasets_refresh_btn = gr.Button("Refresh Datasets", variant="primary")
                 datasets_gallery = gr.HTML()