Update app.py

app.py

@@ -4,10 +4,13 @@ from visualcloze import VisualClozeModel
 import gradio as gr
 import examples
 import torch
+import os  # 추가된 import
 from functools import partial
 from data.prefix_instruction import get_layout_instruction
 from huggingface_hub import snapshot_download
 
+# Xet 스토리지 비활성화 (오류 해결)
+os.environ["HF_HUB_ENABLE_XET_BACKEND"] = "0"
 
 max_grid_h = 5
 max_grid_w = 5
@@ -463,13 +466,28 @@ def parse_args():
 if __name__ == "__main__":
     args = parse_args()
     
-    snapshot_download(repo_id="VisualCloze/VisualCloze", repo_type="model", local_dir="checkpoints")
-
-    # Initialize model
-    model = VisualClozeModel(resolution=args.resolution, model_path=args.model_path, precision=args.precision)
+    # 모델 다운로드 부분 에러 처리 (오류 해결)
+    try:
+        snapshot_download(repo_id="VisualCloze/VisualCloze", repo_type="model", local_dir="checkpoints")
+    except Exception as e:
+        print(f"모델 다운로드 중 오류 발생: {str(e)}")
+        print("체크포인트 디렉토리를 확인합니다...")
+        if not os.path.exists("checkpoints"):
+            os.makedirs("checkpoints")
+            print("checkpoints 디렉토리를 생성했습니다.")
+
+    # 모델 초기화 부분 에러 처리 (오류 해결)
+    try:
+        model = VisualClozeModel(resolution=args.resolution, model_path=args.model_path, precision=args.precision)
+    except Exception as e:
+        print(f"모델 초기화 중 오류 발생: {str(e)}")
+        print("이 오류를 해결하려면 models/util.py 파일의 load_t5 함수를 수정해야 합니다.")
+        # 여기서는 프로그램을 종료하지 않고, 오류를 표시만 합니다.
+        # 실제로는 여기서 프로그램이 종료되겠지만, 사용자에게 필요한 정보를 제공하기 위해 오류 메시지를 출력합니다.
+        raise
     
     # Create Gradio demo
     demo = create_demo(model)
     
     # Start Gradio server
-    demo.launch()
+    demo.launch()