Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 from typing import List, Dict, Union
 import concurrent.futures
 import traceback
+import base64
 
 # 환경 변수에서 토큰 가져오기
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -45,7 +46,7 @@ def capture_thumbnail(space_id: str) -> str:
     try:
         response = requests.get(screenshot_url, headers=get_headers())
         if response.status_code == 200:
-            return response.content
+            return base64.b64encode(response.content).decode('utf-8')
     except requests.RequestException:
         pass
     return None
@@ -119,9 +120,9 @@ def create_ui():
         margin-bottom: 5px !important;
     }
     .thumbnail {
-        max-width: 100px;
-        max-height: 100px;
-        object-fit: contain;
+        width: 100px;
+        height: 100px;
+        object-fit: cover;
     }
     """
 
@@ -149,7 +150,7 @@ def create_ui():
                 for space in formatted_spaces:
                     with gr.Row(elem_classes="space-row") as space_row:
                         if space['thumbnail']:
-                            gr.Image(value=space['thumbnail'], show_label=False, elem_classes="thumbnail")
+                            gr.Image(value=f"data:image/png;base64,{space['thumbnail']}", show_label=False, elem_classes="thumbnail")
                         else:
                             gr.Markdown("No thumbnail")
                         with gr.Column():