r23 commited on
Commit
4ecf70e
·
1 Parent(s): b120863

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from diffusers import StableDiffusionLDM3DPipeline
2
+
3
+ pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d")
4
+ pipe.to("cuda")
5
+
6
+
7
+ prompt ="A picture of some lemons on a table"
8
+ name = "lemons"
9
+
10
+ output = pipe(prompt)
11
+ rgb_image, depth_image = output.rgb, output.depth
12
+ rgb_image[0].save(name+"_ldm3d_rgb.jpg")
13
+ depth_image[0].save(name+"_ldm3d_depth.png")