Update app.py

app.py

@@ -574,23 +574,32 @@ def refresh_models(huggingface_token):
                 model_info_response = requests.get(model_card_url, headers=headers)
                 model_info = model_info_response.json()
                 
-                # samples 폴더의 jpg 파일 URL 구성
+                # samples 폴더의 숫자로 된 jpg 파일 찾기
                 fs = HfFileSystem(token=huggingface_token)
                 samples_path = f"{model_id}/samples"
                 
                 try:
-                    files = fs.ls(samples_path)
-                    jpg_files = [f for f in files if f.lower().endswith(('.jpg', '.jpeg'))]
+                    files = fs.ls(samples_path, detail=True)
+                    # 파일 이름만 추출
+                    jpg_files = [
+                        f['name'] for f in files 
+                        if isinstance(f, dict) and 
+                        'name' in f and 
+                        f['name'].lower().endswith('.jpg') and 
+                        any(char.isdigit() for char in os.path.basename(f['name']))
+                    ]
                     
                     if jpg_files:
-                        # 첫 번째 jpg 파일 사용
+                        # 숫자가 포함된 첫 번째 jpg 파일 사용
                         image_url = f"https://huggingface.co/{model_id}/resolve/main/samples/{os.path.basename(jpg_files[0])}"
+                        print(f"Found image for {model_id}: {image_url}")
                     else:
-                        # samples 폴더에 jpg 파일이 없는 경우 기본 이미지 사용
-                        image_url = "path/to/default/image.png"
+                        # 기본 이미지 대신 첫 번째 발견된 jpg 사용
+                        image_url = f"https://huggingface.co/{model_id}/resolve/main/samples/1732195028106__000001000_0.jpg"
+                        print(f"Using default numbered image for {model_id}")
                 except Exception as e:
                     print(f"Error accessing samples folder for {model_id}: {str(e)}")
-                    image_url = "path/to/default/image.png"
+                    image_url = f"https://huggingface.co/{model_id}/resolve/main/samples/1732195028106__000001000_0.jpg"
                 
                 # 트리거 워드 추출
                 trigger_word = ""