Update app.py

app.py

@@ -18,7 +18,7 @@ current_steps = 25
 SAFETY_CHECKER = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker", torch_dtype=torch.float16)
 
 UPSCALER = DiffusionPipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16)
-UPSCALER.to("cuda")
+UPSCALER.to("cpu")
 UPSCALER.enable_xformers_memory_efficient_attention()
 
 
@@ -40,7 +40,7 @@ class Model:
 
 models = [
  #Model("Stable Diffusion v1-4", "CompVis/stable-diffusion-v1-4"),
- # Model("Stable Diffusion v1-5", "runwayml/stable-diffusion-v1-5"),
+ Model("Stable Diffusion v1-5", "runwayml/stable-diffusion-v1-5"),
  Model("anything-v4.0", "xyn-ai/anything-v4.0"),
 ]
 
@@ -125,7 +125,7 @@ def txt_to_img(
  ).images
 
  pipe.to("cpu")
- torch.cuda.empty_cache()
+ torch.cpu.empty_cache()
 
  return low_res_image[0], up_res_image[0]