Update app.py

app.py

@@ -30,7 +30,10 @@ JS = """function () {
   }
 }"""
 
-# 모델 로드
+# Initialize `pipe` to None globally
+pipe = None
+
+# 모델 로드 시도
 try:
     transformer = FluxTransformer2DModel.from_pretrained("sayakpaul/FLUX.1-merged", torch_dtype=torch.bfloat16)
     if torch.cuda.is_available():
@@ -42,7 +45,6 @@ try:
         print("CUDA is not available. Check your GPU settings.")
 except Exception as e:
     print(f"Failed to load the model: {e}")
-    pipe = None
 
 # 이미지 생성 함수
 @spaces.GPU(duration=120)