ldm3d-space / app.py
r23's picture
Update app.py
864dbf5
import os
os.environ["HF_HOME"] = "/tmp/hf_cache"
from diffusers import StableDiffusionLDM3DPipeline
pipe_ldm3d = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d")
prompt = "A picture of a castle in the mountains"
output = pipe_ldm3d(prompt)
obj_file = output.obj
obj_file[0].save("castle_ldm3d.obj")