Update app.py

app.py

@@ -85,7 +85,7 @@ neg_prompt_2 = " 'non-photorealistic':1.5, 'unrealistic skin','unattractive face
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device("cpu"))
+upscaler = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
 
 def load_and_prepare_model():
     vaeX=AutoencoderKL.from_pretrained("ford442/stable-diffusion-3.5-large-fp32", safety_checker=None, use_safetensors=True, subfolder='vae', low_cpu_mem_usage=False, torch_dtype=torch.float32, token=True)
@@ -232,7 +232,7 @@ def generate(
         output_image_file = f"Large_Lora_L2_{seed}.png"
         #image.save(output_image_file,optimize=False,compress_level=0)
         #upload_to_ftp(output_image_file) 
-        upscaler.to(torch.device('cuda'))
+        #upscaler.to(torch.device('cuda'))
         with torch.no_grad():
             upscale2 = upscaler(image, tiling=True, tile_width=256, tile_height=256)
         #timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")