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 |
```
|