Update app.py

app.py

@@ -28,7 +28,7 @@ def infer(prompt, samples, steps, scale, seed):
     generator = torch.Generator(device=device).manual_seed(seed)
     
     #If you are running locally with CPU, you can remove the `with autocast("cuda")`
-    with autocast("cuda"):
+    with autocast("cpu"):
         images_list = pipe(
             [prompt] * samples,
             num_inference_steps=steps,