Update README.md
Browse files
README.md
CHANGED
|
@@ -36,7 +36,8 @@ pipe_ldm3d.to("cuda")
|
|
| 36 |
prompt ="A picture of some lemons on a table"
|
| 37 |
name = "lemons"
|
| 38 |
|
| 39 |
-
|
|
|
|
| 40 |
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
|
| 41 |
depth_image[0].save(name+"_ldm3d_depth.png")
|
| 42 |
```
|
|
|
|
| 36 |
prompt ="A picture of some lemons on a table"
|
| 37 |
name = "lemons"
|
| 38 |
|
| 39 |
+
output = pipe_ldm3d(prompt)
|
| 40 |
+
rgb_image, depth_image = output.rgb, output.depth
|
| 41 |
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
|
| 42 |
depth_image[0].save(name+"_ldm3d_depth.png")
|
| 43 |
```
|