Update app.py

app.py

@@ -71,25 +71,57 @@ def translate_to_english(text: str) -> str:
         print(f"Translation error: {str(e)}")
         return text
         
-# Eric cat LoRA 가중치 로드
+print("Initializing FLUX pipeline...")
 try:
+    pipe = FluxPipeline.from_pretrained(
+        "black-forest-labs/FLUX.1-dev",
+        torch_dtype=torch.float16,
+        use_auth_token=HF_TOKEN,
+        safety_checker=None  # 안전성 검사 비활성화
+    )
+    print("FLUX pipeline initialized successfully")
+    
+    # 메모리 최적화 설정
+    pipe.enable_attention_slicing(slice_size="auto")
+    pipe.enable_model_cpu_offload()  # CPU 오프로딩 활성화
+    print("Pipeline optimization settings applied")
+
+except Exception as e:
+    print(f"Error initializing FLUX pipeline: {str(e)}")
+    raise
+
+# LoRA 가중치 로드 부분 수정
+print("Loading LoRA weights...")
+try:
+    # LoRA 파일 경로 확인
     lora_path = hf_hub_download(
-        "openfree/myt-flux-fantasy",
-        "myt-flux-fantasy.safetensors",
+        repo_id="openfree/myt-flux-fantasy",
+        filename="myt-flux-fantasy.safetensors",  # 정확한 파일명 확인 필요
         use_auth_token=HF_TOKEN
     )
+    print(f"LoRA weights downloaded to: {lora_path}")
+    
+    # LoRA 가중치 로드
     pipe.load_lora_weights(lora_path)
     pipe.fuse_lora(lora_scale=0.125)
+    print("LoRA weights loaded and fused successfully")
+
 except Exception as e:
     print(f"Error loading LoRA weights: {str(e)}")
+    print(f"Full error details: {repr(e)}")
     raise ValueError("Failed to load LoRA weights. Please check your HF_TOKEN and model access.")
 
-# GPU로 이동
+# GPU 이동 부분 수정
 if torch.cuda.is_available():
     try:
+        print("Moving pipeline to GPU...")
         pipe = pipe.to("cuda:0")
+        print("Pipeline successfully moved to GPU")
+        print(f"Current device: {pipe.device}")
     except Exception as e:
         print(f"Warning: Could not move pipeline to CUDA: {str(e)}")
+        print("Falling back to CPU")
+
 
 # 저장 디렉토리 설정
 SAVE_DIR = "saved_images"