Update README.md
Browse files
README.md
CHANGED
|
@@ -45,6 +45,7 @@ import torch
|
|
| 45 |
pipe = TerrainDiffusionPipeline.from_pretrained("./weights", torch_dtype=torch.float16)
|
| 46 |
pipe.to("cuda");
|
| 47 |
|
|
|
|
| 48 |
image,dem = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
|
| 49 |
```
|
| 50 |
|
|
|
|
| 45 |
pipe = TerrainDiffusionPipeline.from_pretrained("./weights", torch_dtype=torch.float16)
|
| 46 |
pipe.to("cuda");
|
| 47 |
|
| 48 |
+
prompt = "A sentinel-2 image of montane forests and mountains in Mexico in August"
|
| 49 |
image,dem = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)
|
| 50 |
```
|
| 51 |
|