Update app.py

app.py

@@ -39,14 +39,37 @@ os.environ["TRANSFORMERS_CACHE"] = cache_path
 os.environ["HF_HUB_CACHE"] = cache_path
 os.environ["HF_HOME"] = cache_path
 
+
+# Hugging Face 토큰 설정
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    print("Warning: HF_TOKEN not found in environment variables")
+
 # FLUX 모델 초기화
 if not path.exists(cache_path):
     os.makedirs(cache_path, exist_ok=True)
 
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-pipe.load_lora_weights(hf_hub_download("ByteDance/Hyper-SD", "Hyper-FLUX.1-dev-8steps-lora.safetensors"))
-pipe.fuse_lora(lora_scale=0.125)
-pipe.to(device="cuda", dtype=torch.bfloat16)
+try:
+    pipe = FluxPipeline.from_pretrained(
+        "black-forest-labs/FLUX.1-dev",
+        torch_dtype=torch.bfloat16,
+        use_auth_token=HF_TOKEN  # Hugging Face 토큰 추가
+    )
+    pipe.load_lora_weights(
+        hf_hub_download(
+            "ByteDance/Hyper-SD",
+            "Hyper-FLUX.1-dev-8steps-lora.safetensors",
+            token=HF_TOKEN  # Hugging Face 토큰 추가
+        )
+    )
+    pipe.fuse_lora(lora_scale=0.125)
+    pipe.to(device="cuda", dtype=torch.bfloat16)
+    print("Successfully initialized FLUX model with authentication")
+except Exception as e:
+    print(f"Error initializing FLUX model: {str(e)}")
+    pipe = None
+    
+
 
 # 이미지 생성 함수 추가
 @spaces.GPU