Update app.py

app.py

@@ -63,12 +63,13 @@ controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
 )
 
+# Upscale 파이프라인 설정 (기존 pipe 재사용)
 # Upscale 파이프라인 설정 (기존 pipe 재사용)
 pipe_upscale = FluxControlNetPipeline(
     vae=pipe.vae,
     text_encoder=pipe.text_encoder,
     tokenizer=pipe.tokenizer,
-    unet=pipe.unet,
+    transformer=pipe.transformer,  # FLUX 모델은 transformer를 사용할 수 있습니다
     scheduler=pipe.scheduler,
     safety_checker=pipe.safety_checker,
     feature_extractor=pipe.feature_extractor,