forgot rgb type

app.py

@@ -11,7 +11,7 @@ def get_depth(rgb):
     rgb = rgb.convert("RGB")
     depth = depth_estimator.get_depth(rgb)
 
-    h, w, _ = rgb.shape
+    w, h = rgb.size
     grid = np.mgrid[0:h, 0:w].transpose(1, 2, 0
                                         ).reshape(-1, 2)[..., ::-1]
     flat_grid = grid[:, 1] * w + grid[:, 0]
@@ -37,7 +37,7 @@ def get_depth(rgb):
     ))
 
     tf = tempfile.NamedTemporaryFile(suffix=".obj").name
-    save_obj(positions, rgb.reshape(-1, 3), faces, tf)
+    save_obj(positions, np.asarray(rgb).reshape(-1, 3), faces, tf)
 
     return rgb, (depth.clip(0, 64) * 1024).astype("uint16"), tf
 
@@ -56,6 +56,6 @@ gr.Interface(fn=get_depth, inputs=[
 ], outputs=[
     gr.components.Image(type="pil", label="image"),
     gr.components.Image(type="numpy", label="depth"),
-    gr.components.Model3D()
+    gr.components.Model3D(label="3d model")
 
 ]).launch(share=True)