Views
No views yet
1from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
2import torch
3from PIL import Image
4
5img = Image.open("osm_tile_18_42048_101323.jpeg")
6
7controlnet = ControlNetModel.from_pretrained("MVRL/GeoSynth-OSM")
8
9pipe = StableDiffusionControlNetPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", controlnet=controlnet)
10pipe = pipe.to("cuda:0")
11
12# generate image
13generator = torch.manual_seed(10345340)
14image = pipe(
15 "Satellite image features a city neighborhood",
16 generator=generator,
17 image=img,
18).images[0]
19
20image.save("generated_city.jpg")