Update app.py

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline, StableDiffusionDepth2ImgPipeline
+from PIL import Image
+import numpy as np
+import open3d as o3d
+
+# Initialize the pipelines
+sd_pipeline = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", torch_dtype="float16")
+sd_pipeline.to("cuda")
+depth_pipeline = StableDiffusionDepth2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-depth2img", torch_dtype="float16")
+depth_pipeline.to("cuda")
+
+def generate_images(prompt):
+    # Generate the base image
+    image = sd_pipeline(prompt).images[0]
+    # Generate the depth map
+    depth_image = depth_pipeline(prompt=prompt, image=image).images[0]
+    return image, depth_image
+
+# Define Gradio interface
+iface = gr.Interface(
+    fn=generate_images,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=[
+        gr.Image(label="Bas-relief Image"),
+        gr.Image(label="Depth Map Image")
+    ],
+    title="Text to Bas-Relief and Depth Map Generator"
+)
+
+iface.launch()