Update app.py

app.py

@@ -688,7 +688,7 @@ def generate_map(location_names):
     map_html = m._repr_html_()
     return map_html
 
-
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", torch_dtype=torch.float16)
 pipe.to(device)