Update app.py

app.py

@@ -20,7 +20,7 @@ from RealESRGAN import RealESRGAN
 import gradio as gr
 from gradio_imageslider import ImageSlider
 
-USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
+USE_TORCH_COMPILE = True
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -165,7 +165,6 @@ def create_hdr_effect(original_image, hdr):
 
 lazy_pipe = LazyLoadPipeline()
 lazy_pipe.load()
-lazy_pipe = lazy_pipe.to('cuda')
 
 def prepare_image(input_image, resolution, hdr):
     condition_image = resize_and_upscale(input_image, resolution)