Update app.py

app.py

@@ -23,6 +23,11 @@ os.makedirs(TMP_DIR, exist_ok=True)
 def initialize_models():
     global pipeline, translator, flux_pipe
     
+    # Hugging Face 토큰 확인
+    hf_token = os.getenv("HF_TOKEN")
+    if not hf_token:
+        raise ValueError("HF_TOKEN environment variable is not set. Please set your Hugging Face token.")
+    
     # Trellis 파이프라인 초기화
     pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
     pipeline.cuda()
@@ -30,9 +35,16 @@ def initialize_models():
     # 번역기 초기화
     translator = translation_pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
     
-    # Flux 파이프라인 초기화
-    flux_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-    flux_pipe.load_lora_weights("gokaygokay/Flux-Game-Assets-LoRA-v2")
+    # Flux 파이프라인 초기화 - token 추가
+    flux_pipe = FluxPipeline.from_pretrained(
+        "black-forest-labs/FLUX.1-dev", 
+        torch_dtype=torch.bfloat16,
+        use_auth_token=hf_token  # Hugging Face 토큰 적용
+    )
+    flux_pipe.load_lora_weights(
+        "gokaygokay/Flux-Game-Assets-LoRA-v2",
+        use_auth_token=hf_token  # LoRA 가중치 로드시에도 토큰 적용
+    )
     flux_pipe.fuse_lora(lora_scale=1.0)
     flux_pipe.to(device="cuda", dtype=torch.bfloat16)