Update app.py

app.py

@@ -867,28 +867,37 @@ def step1_gpu_refine(
 
         # ---- (옵션) 리드로우(img2img)
     if do_redraw_flag:
-        from diffusers import StableDiffusionImg2ImgPipeline
-        pr = StableDiffusionImg2ImgPipeline.from_pretrained(
-            "runwayml/stable-diffusion-v1-5", torch_dtype=dtype,
-            safety_checker=None, feature_extractor=None
-        )
-        pr = _disable_safety(pr)
-        if dev == "cuda": pr.to("cuda")
-        img_for = _resize_to_multiple(img.convert("RGB"), 8, 640)
-        img = pr(
-            prompt="clean anime illustration, sharp lines, flat colors, plain white background",
-            negative_prompt="glitch, mosaic, text, watermark, noisy",
-            image=img_for,
-            strength=float(max(0.30, min(0.45, float(redraw_strength)))),
-            guidance_scale=float(max(6.5, min(9.0, float(redraw_guidance)))),
-            num_inference_steps=int(max(14, min(28, int(redraw_steps)))),
-        ).images[0].convert("RGBA")
-        
-        img.save(OUT/"step1"/"dbg_05_after_redraw.png")   # ✅ 들여쓰기 맞춤
-        logs.append("img2img 리드로우 적용")              # ✅ 들여쓰기 맞춤
+        try:
+            from diffusers import StableDiffusionImg2ImgPipeline
+            pr = StableDiffusionImg2ImgPipeline.from_pretrained(
+                "runwayml/stable-diffusion-v1-5",
+                torch_dtype=dtype,
+                safety_checker=None,
+                feature_extractor=None
+            )
+    
+            pr = _disable_safety(pr)
+            if dev == "cuda":
+                pr.to("cuda")
+    
+            img_for = _resize_to_multiple(img.convert("RGB"), 8, 640)
+    
+            img = pr(
+                prompt="clean anime illustration, sharp lines, flat colors, plain white background",
+                negative_prompt="glitch, mosaic, text, watermark, noisy",
+                image=img_for,
+                strength=float(max(0.30, min(0.45, float(redraw_strength)))),
+                guidance_scale=float(max(6.5, min(9.0, float(redraw_guidance)))),
+                num_inference_steps=int(max(14, min(28, int(redraw_steps))))
+            ).images[0].convert("RGBA")
+    
+            img.save(OUT/"step1"/"dbg_05_after_redraw.png")
+            logs.append("img2img 리드로우 적용")
+    
         except Exception as e:
             logs.append(f"img2img 리드로우 실패: {e}")
 
+
     # ---- 저장 & 프리뷰
     out_path = _save_png(img, OUT / "step1" / "input_preprocessed.png")
     try: