Update app.py

app.py

@@ -59,7 +59,7 @@ pipe_i2i = AutoPipelineForImage2Image.from_pretrained(
     torch_dtype=dtype
 )
 
-# Load controlnet model for upscaling
+# 업스케일링을 위한 ControlNet 모델 로드
 controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=dtype
 ).to(device)
@@ -72,10 +72,13 @@ pipe_controlnet = FluxControlNetPipeline(
     tokenizer_2=pipe.tokenizer_2,
     transformer=pipe.transformer,  # unet 대신 transformer 사용
     controlnet=controlnet,
-    scheduler=pipe.scheduler,
-    torch_dtype=dtype
+    scheduler=pipe.scheduler
+    # 'safety_checker'와 'feature_extractor'는 제거됨
 ).to(device)
 
+
+
+
 MAX_SEED = 2**32 - 1
 MAX_PIXEL_BUDGET = 1024 * 1024