jt5d's picture
Update app.py
8ffeb4b
raw
history blame
622 Bytes
#import gradio as gr
#gr.Interface.load("models/Intel/ldm3d-pano").launch()
!pip install --upgrade diffusers[torch]
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")