estellea commited on
Commit
0763a6a
1 Parent(s): 6d2dd3e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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
- rgb_image, depth_image = pipe_ldm3d(prompt).images
 
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
  ```