LFS file download and load in the code

app.py

@@ -16,6 +16,8 @@ from typing import List
 import torch
 import os
 import io
+import warnings
+import requests
 from transformers import AutoTokenizer
 import numpy as np
 from utils_mask import get_mask_location
@@ -26,9 +28,15 @@ from preprocess.openpose.run_openpose import OpenPose
 from detectron2.data.detection_utils import convert_PIL_to_numpy,_apply_exif_orientation
 from torchvision.transforms.functional import to_pil_image
 import pillow_heif  # HEIC 이미지 처리용 (아이폰 촬영 사진 포맷)
-import requests
 from urllib.parse import urlparse
 
+# SSL 경고 억제
+warnings.filterwarnings("ignore", message=".*OpenSSL.*")
+warnings.filterwarnings("ignore", category=UserWarning, module="urllib3")
+
+# requests 세션 설정
+session = requests.Session()
+session.verify = False  # SSL 검증 비활성화 (개발 환경용)
 
 def pil_to_binary_mask(pil_image, threshold=0):
     np_image = np.array(pil_image)
@@ -166,7 +174,7 @@ def preprocess_image(image):
 # URL에서 이미지 가져오기 함수
 def load_image_from_url(url):
     try:
-        response = requests.get(url, stream=True, timeout=10)
+        response = session.get(url, stream=True, timeout=10)
         response.raise_for_status()  # HTTP 오류 확인
         
         # 이미지 다운로드
@@ -234,7 +242,7 @@ def download_densepose_model():
     
     try:
         print(f"Downloading DensePose model from {download_url}")
-        response = requests.get(download_url, stream=True, timeout=300)
+        response = session.get(download_url, stream=True, timeout=300)
         response.raise_for_status()
         
         with open(model_path, 'wb') as f:
@@ -250,7 +258,7 @@ def download_densepose_model():
         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 = session.get(fallback_url, stream=True, timeout=300)
             response.raise_for_status()
             
             with open(model_path, 'wb') as f:
@@ -455,7 +463,7 @@ with image_blocks as demo:
     # is_checked = gr.Number(value=True)
 
     # 이미지 업로드 시 전처리
-    imgs.upload(
+    imgs.change(
         fn=preprocess_image,
         inputs=imgs,
         outputs=imgs,  # 전처리된 이미지를 ImageEditor에 다시 표시
@@ -486,10 +494,20 @@ with image_blocks as demo:
         api_name='tryon'
     )
 
-image_blocks.launch()
-
 # 앱 시작 시 DensePose 모델 미리 다운로드
-if __name__ == "__main__":
-    print("Initializing DensePose model...")
+print("Initializing DensePose model...")
+try:
     download_densepose_model()
     print("DensePose model initialization completed.")
+except Exception as e:
+    print(f"Warning: Could not download DensePose model: {e}")
+    print("The model will be downloaded when needed during inference.")
+
+# 앱 실행
+if __name__ == "__main__":
+    try:
+        print("Starting GENAI-VTON application...")
+        image_blocks.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    except Exception as e:
+        print(f"Error starting the application: {e}")
+        print("Please check if all required dependencies are installed.")