Spaces:
Runtime error
Runtime error
File size: 581 Bytes
c361e9a 5f2ad5e c361e9a 5f2ad5e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
#import gradio as gr
#gr.Interface.load("models/Intel/ldm3d-pano").launch()
from diffusers import StableDiffusionLDM3DPipeline
pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d-pano")
pipe.to("cuda")
prompt ="360 view of a photorealistic 4K futuristic cityscape"
name = "bedroom_pano"
output = pipe(
prompt,
width=1024,
height=512,
guidance_scale=7.0,
num_inference_steps=50,
)
rgb_image, depth_image = output.rgb, output.depth
rgb_image[0].save(name+"_ldm3d_rgb.jpg")
depth_image[0].save(name+"_ldm3d_depth.png")
|