User LFS download

app.py

@@ -207,11 +207,64 @@ def process_url_image(url):
     img = load_image_from_url(url)
     if img is None:
         print("Failed to load image from URL")
-        return None    
-   
+        return None
+        
     return preprocess_image(img)
 
 
+# Hugging Face LFS OID를 사용하여 densepose 모델 다운로드
+def download_densepose_model():
+    """
+    Hugging Face LFS OID를 사용하여 densepose 모델을 다운로드합니다.
+    OID: b8a7382001b16e453bad95ca9dbc68ae8f2b839b304cf90eaf5c27fbdb4dae91
+    """
+    model_path = './ckpt/densepose/model_final_162be9.pkl'
+    
+    # 모델 파일이 이미 존재하는지 확인
+    if os.path.exists(model_path):
+        print(f"DensePose model already exists at {model_path}")
+        return model_path
+    
+    # 디렉토리 생성
+    os.makedirs('./ckpt/densepose', exist_ok=True)
+    
+    # Hugging Face LFS OID를 사용한 다운로드 URL
+    oid = "b8a7382001b16e453bad95ca9dbc68ae8f2b839b304cf90eaf5c27fbdb4dae91"
+    download_url = f"https://huggingface.co/datasets/hf-internal-testing/fixtures/resolve/{oid}/model_final_162be9.pkl"
+    
+    try:
+        print(f"Downloading DensePose model from {download_url}")
+        response = requests.get(download_url, stream=True, timeout=300)
+        response.raise_for_status()
+        
+        with open(model_path, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+        
+        print(f"DensePose model downloaded successfully to {model_path}")
+        return model_path
+        
+    except Exception as e:
+        print(f"Error downloading DensePose model: {e}")
+        # 대체 URL 시도
+        fallback_url = "https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/165712039/model_final_162be9.pkl"
+        try:
+            print(f"Trying fallback URL: {fallback_url}")
+            response = requests.get(fallback_url, stream=True, timeout=300)
+            response.raise_for_status()
+            
+            with open(model_path, 'wb') as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            
+            print(f"DensePose model downloaded successfully from fallback URL to {model_path}")
+            return model_path
+            
+        except Exception as e2:
+            print(f"Error downloading from fallback URL: {e2}")
+            return None
+
+
 @spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,denoise_steps,seed, is_checked_crop):
     device = "cuda"
@@ -256,7 +309,13 @@ def start_tryon(dict,garm_img,garment_des,is_checked,denoise_steps,seed, is_chec
      
     
 
-    args = apply_net.create_argument_parser().parse_args(('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', './ckpt/densepose/model_final_162be9.pkl', 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda'))
+    # DensePose 모델 다운로드 및 경로 설정
+    densepose_model_path = download_densepose_model()
+    if densepose_model_path is None:
+        print("Failed to download DensePose model, using default path")
+        densepose_model_path = './ckpt/densepose/model_final_162be9.pkl'
+    
+    args = apply_net.create_argument_parser().parse_args(('show', './configs/densepose_rcnn_R_50_FPN_s1x.yaml', densepose_model_path, 'dp_segm', '-v', '--opts', 'MODEL.DEVICE', 'cuda'))
     # verbosity = getattr(args, "verbosity", None)
     pose_img = args.func(args,human_img_arg)    
     pose_img = pose_img[:,:,::-1]    
@@ -428,3 +487,9 @@ with image_blocks as demo:
     )
 
 image_blocks.launch()
+
+# 앱 시작 시 DensePose 모델 미리 다운로드
+if __name__ == "__main__":
+    print("Initializing DensePose model...")
+    download_densepose_model()
+    print("DensePose model initialization completed.")