Update app.py

app.py

@@ -1,3 +1,25 @@
 import gradio as gr
 
-gr.Interface.load("models/Intel/ldm3d-pano").launch()
+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")