Update app.py

app.py

@@ -100,9 +100,19 @@ def create_3d_obj(rgb_image, depth_image, image_path, depth=10):
     return gltf_path
 
 
-title = "zero-shot depth estimation with DPT + 3D Point Cloud"
-description = "DPT model to predict the depth of an image and 3D Point Cloud to create a 3D object."
-examples = [["examples/" + img] for img in os.listdir("examples/")]
+title = "Demo: zero-shot depth estimation with DPT + 3D Point Cloud"
+description = "This demo is a variation from the original <a href='https://huggingface.co/spaces/nielsr/dpt-depth-estimation' target='_blank'>DPT Demo</a>. It uses the DPT model to predict the depth of an image and then uses 3D Point Cloud to create a 3D object."
+
+# Add both image and model examples
+examples = [
+    ["examples/" + img] for img in os.listdir("files/")
+] + [
+    [os.path.join(os.path.dirname(__file__), "files/model1.glb")],
+    [os.path.join(os.path.dirname(__file__), "files/model2.glb")],
+    [os.path.join(os.path.dirname(__file__), "files/model3.glb")],
+    [os.path.join(os.path.dirname(__file__), "files/model4.glb")],
+    ["https://huggingface.co/datasets/dylanebert/3dgs/resolve/main/bonsai/bonsai-7k-mini.splat"],
+]
 
 iface = gr.Interface(fn=process_image,
                      inputs=[gr.Image(
@@ -117,6 +127,10 @@ iface = gr.Interface(fn=process_image,
                      allow_flagging="never",
                      cache_examples=False)
 
+if __name__ == "__main__":
+    iface.launch()
+
+