r23 commited on
Commit
6cae4b7
·
1 Parent(s): 3e23b21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -9
app.py CHANGED
@@ -1,13 +1,9 @@
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")
 
1
  from diffusers import StableDiffusionLDM3DPipeline
2
 
3
+ pipe_ldm3d = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d")
 
4
 
5
+ prompt = "A picture of a castle in the mountains"
6
 
7
+ output = pipe_ldm3d(prompt)
8
+ obj_file = output.obj
9
+ obj_file[0].save("castle_ldm3d.obj")