Update app.py

app.py

@@ -50,6 +50,13 @@ def capture_thumbnail(space_id: str) -> bytes:
         pass
     return None
 
+def get_hardware_requirements(space: Dict) -> str:
+    sdk_version = space.get('sdk', {}).get('hf', {}).get('version')
+    if sdk_version and int(sdk_version.split('.')[0]) >= 3:
+        return space.get('sdk', {}).get('hardware', 'Unknown')
+    else:
+        return 'CPU' if space.get('host_requirements', {}).get('hardware', {}).get('gpu') is False else 'GPU'
+
 def format_space(space: Dict) -> Dict:
     space_id = space.get('id', 'Unknown')
     space_name = space_id.split('/')[-1] if '/' in space_id else space_id
@@ -62,6 +69,7 @@ def format_space(space: Dict) -> Dict:
     space_url = f"https://huggingface.co/spaces/{space_id}"
     
     thumbnail = capture_thumbnail(space_id)
+    hardware = get_hardware_requirements(space)
     
     return {
         "id": space_id,
@@ -69,7 +77,8 @@ def format_space(space: Dict) -> Dict:
         "author": space_author,
         "likes": space_likes,
         "url": space_url,
-        "thumbnail": thumbnail
+        "thumbnail": thumbnail,
+        "hardware": hardware
     }
 
 def format_spaces(spaces: Union[List[Dict], str]) -> List[Dict]:
@@ -126,6 +135,10 @@ def create_ui():
         height: 100px;
         object-fit: cover;
     }
+    .hardware-info {
+        font-size: 12px;
+        color: #666;
+    }
     """
 
     def on_select(space):
@@ -135,6 +148,7 @@ def create_ui():
             info = f"선택된 Space: {space['name']} (ID: {space['id']})\n"
             info += f"Author: {space['author']}\n"
             info += f"Likes: {space['likes']}\n"
+            info += f"Hardware: {space['hardware']}\n"
             info += f"URL: {space['url']}\n\n"
             info += f"요약:\n{summary}"
             return info, app_content
@@ -157,6 +171,7 @@ def create_ui():
                             gr.Image(value="https://huggingface.co/front/assets/huggingface_logo-noborder.svg", shape=(100, 100), show_label=False, elem_classes="thumbnail")
                         with gr.Column():
                             gr.Markdown(f"{space['name']} by {space['author']} (Likes: {space['likes']})", elem_classes="space-info")
+                            gr.Markdown(f"Hardware: {space['hardware']}", elem_classes="hardware-info")
                             button = gr.Button("클릭", elem_classes="minimal-button")
                     space_rows.append((space_row, button, space))