Update app.py

app.py

@@ -69,26 +69,22 @@ print("Available attributes in FLUX pipeline:", pipe.__dict__.keys())
 pipe_upscale = FluxControlNetPipeline(
     vae=pipe.vae,
     text_encoder=pipe.text_encoder,
+    text_encoder_2=pipe.text_encoder_2,
     tokenizer=pipe.tokenizer,
+    tokenizer_2=pipe.tokenizer_2,
     transformer=pipe.transformer,
     scheduler=pipe.scheduler,
     controlnet=controlnet
 )
 
-# 사용 가능한 경우에만 추가 속성 설정
-if hasattr(pipe, 'feature_extractor'):
-    pipe_upscale.feature_extractor = pipe.feature_extractor
-
-# 모든 파이프라인을 device로 이동
-pipe_upscale.to(device)
+# 추가 속성 설정 (있는 경우에만)
+if hasattr(pipe, 'image_processor'):
+    pipe_upscale.image_processor = pipe.image_processor
 
 # 모든 파이프라인을 device로 이동
-pipe_i2i.to(device)
-controlnet.to(device)
 pipe_upscale.to(device)
 
 
-
 MAX_SEED = 2**32 - 1
 MAX_PIXEL_BUDGET = 1024 * 1024