fix device

app_upscale.py

@@ -4,9 +4,13 @@ from io import BytesIO
 from diffusers import StableDiffusionUpscalePipeline
 import torch
 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+print('f{device} is available')
+
 model_id = "stabilityai/stable-diffusion-x4-upscaler"
 upscale_pipe = StableDiffusionUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-upscale_pipe = upscale_pipe.to("cuda")
+upscale_pipe = upscale_pipe.to(device)
 
 DEFAULT_SRC_PROMPT = "a person with pefect face"