Remove download logic with huggingface OID cause Token information issue

app.py

@@ -220,11 +220,10 @@ def process_url_image(url):
     return preprocess_image(img)
 
 
-# Hugging Face LFS OID를 사용하여 densepose 모델 다운로드
+# DensePose 모델 다운로드
 def download_densepose_model():
     """
-    Hugging Face LFS OID를 사용하여 densepose 모델을 다운로드합니다.
-    OID: b8a7382001b16e453bad95ca9dbc68ae8f2b839b304cf90eaf5c27fbdb4dae91
+    DensePose 모델을 안정적인 공개 URL에서 다운로드합니다.
     """
     model_path = './ckpt/densepose/model_final_162be9.pkl'
     
@@ -236,41 +235,32 @@ def download_densepose_model():
     # 디렉토리 생성
     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"
+    # 안정적인 다운로드 URL 목록
+    download_urls = [
+        "https://dl.fbaipublicfiles.com/densepose/densepose_rcnn_R_50_FPN_s1x/165712039/model_final_162be9.pkl",
+        "https://github.com/facebookresearch/densepose/releases/download/v1.0/model_final_162be9.pkl"
+    ]
     
-    try:
-        print(f"Downloading DensePose model from {download_url}")
-        response = session.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"
+    for i, download_url in enumerate(download_urls, 1):
         try:
-            print(f"Trying fallback URL: {fallback_url}")
-            response = session.get(fallback_url, stream=True, timeout=300)
+            print(f"Attempting download {i}/{len(download_urls)} from: {download_url}")
+            response = session.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 from fallback URL to {model_path}")
+            print(f"DensePose model downloaded successfully from source {i} to {model_path}")
             return model_path
             
-        except Exception as e2:
-            print(f"Error downloading from fallback URL: {e2}")
-            return None
+        except Exception as e:
+            print(f"Error downloading from source {i}: {e}")
+            continue
+    
+    print("Failed to download DensePose model from all sources")
+    print("Please manually download the model file and place it in ./ckpt/densepose/model_final_162be9.pkl")
+    return None
 
 
 @spaces.GPU