Update app.py

app.py

@@ -24,6 +24,15 @@ from transformers import pipeline
 from diffusers import FluxPipeline
 from PIL import Image
 import numpy as np
+from huggingface_hub import login
+
+# Hugging Face 토큰 설정 및 로그인
+hf_token = os.getenv("HF_TOKEN")
+if hf_token:
+    login(token=hf_token)
+else:
+    print("Warning: HF_TOKEN not found in environment variables. You may encounter authentication issues.")
+
 
 def download_model():
     REPO_ID = 'Doubiiu/DynamiCrafter_1024'
@@ -50,10 +59,15 @@ model = model.cuda()
 # 번역 모델 초기화
 translator = 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 파이프라인 초기화 부분 수정
+flux_pipe = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev",
+    torch_dtype=torch.bfloat16,
+    use_auth_token=hf_token  # 토큰을 사용하여 인증
+)
 flux_pipe.enable_model_cpu_offload()
 
+
 def generate_image_from_text(prompt, seed=0):
     generator = torch.Generator("cpu").manual_seed(seed)
     image = flux_pipe(