Update app.py

app.py

@@ -5,7 +5,6 @@ import requests
 from typing import List, Dict, Union
 import concurrent.futures
 import traceback
-import base64
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -41,12 +40,12 @@ def get_most_liked_spaces(limit: int = 100) -> Union[List[Dict], str]:
     except ValueError as e:
         return f"JSON decoding error: {str(e)}"
 
-def capture_thumbnail(space_id: str) -> str:
+def capture_thumbnail(space_id: str) -> bytes:
     screenshot_url = f"https://huggingface.co/spaces/{space_id}/thumbnail.png"
     try:
         response = requests.get(screenshot_url, headers=get_headers())
         if response.status_code == 200:
-            return base64.b64encode(response.content).decode('utf-8')
+            return response.content
     except requests.RequestException:
         pass
     return None
@@ -153,9 +152,9 @@ def create_ui():
                 for space in formatted_spaces:
                     with gr.Row(elem_classes="space-row") as space_row:
                         if space['thumbnail']:
-                            gr.Image(value=f"data:image/png;base64,{space['thumbnail']}", show_label=False, elem_classes="thumbnail")
+                            gr.Image(value=space['thumbnail'], shape=(100, 100), show_label=False, elem_classes="thumbnail")
                         else:
-                            gr.Image(value="https://huggingface.co/front/assets/huggingface_logo-noborder.svg", show_label=False, elem_classes="thumbnail")
+                            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")
                             button = gr.Button("클릭", elem_classes="minimal-button")