Update app.py

app.py

@@ -183,19 +183,25 @@ def create_3d_obj(foreground, depth_image, background, image_path, depth=10):
     return str(gltf_path)
 
 # Create Gradio interface
-interface = gr.Interface(
+title = "Demo: Depth Estimation & 3D Reconstruction with DPT + Point Cloud"
+description = "This demo allows users to provide an RGB image and optionally a depth map to create a 3D object. If no depth map is provided, the DPT model will generate it."
+examples = [["examples/" + img] for img in os.listdir("examples/")]
+
+iface = gr.Interface(
     fn=process_image,
     inputs=[
-        gr.inputs.Image(label="Upload Image"),
-        gr.inputs.File(label="Upload Depth Map (optional)", type="file")
+        gr.Image(type="filepath", label="Input Image"),
+        gr.Image(type="filepath", label="Input Depth Map (optional)"),
     ],
     outputs=[
-        gr.outputs.Image(label="Processed Image"),
-        gr.outputs.File(label="3D Model Output (GLTF)")
+        gr.Image(label="Predicted Depth", type="pil"),
+        gr.Model3D(label="3D Mesh Reconstruction", clear_color=[1.0, 1.0, 1.0, 1.0]),
+        gr.File(label="3D gLTF File"),
     ],
-    title="Depth Estimation and 3D Model Generation",
-    description="Upload an image and optionally a depth map to generate a 3D model."
+    title=title,
+    description=description,
+    examples=examples,
+    allow_flagging="never",
+    cache_examples=False,
 )
-
-if __name__ == "__main__":
-    interface.launch()
+iface.launch(debug=True, show_api=True, share=True)