Update app.py

app.py

@@ -24,12 +24,12 @@ def generate_and_display_images(model_selection, scenery, style, height, width,
 
     generated_images = []
     if model_selection == "dreamlike-art/dreamlike-photoreal-2.0":
-        model = StableDiffusionPipeline.from_pretrained(model_selection, torch_dtype=torch.float16).to("cuda")
+        model = StableDiffusionPipeline.from_pretrained(model_selection, torch_dtype=torch.float16).to("cpu")
         for _ in range(num_images):
             image = model(prompt=prompt, num_inference_steps=n_steps, guidance_scale=guidance_scale, negative_prompt=negative_prompt, height=height, width=width).images[0]
             generated_images.append(image)
     else:
-        base = DiffusionPipeline.from_pretrained(model_selection, torch_dtype=torch.float16, use_auth_token=True).to("cuda")
+        base = DiffusionPipeline.from_pretrained(model_selection, torch_dtype=torch.float16, use_auth_token=True).to("cpu")
         for _ in range(num_images):
             if "refiner" in model_selection:
                 image_latent = base(prompt=prompt, num_inference_steps=n_steps, denoising_end=high_noise_frac, output_type="latent").images