Update app.py

app.py

@@ -229,7 +229,7 @@ def rgb2gray(rgb):
     return np.dot(rgb[...,:3], [0.333, 0.333, 0.333])
 
 def get_mesh(image, depth, fnum):
-    gdepth = rgb2gray(depth[fnum])
+    gdepth = rgb2gray(depth[fnum][0])
     print('depth to gray - ok')
     points = pano_depth_to_world_points(gdepth, 1)
     pts3d = points[0]
@@ -243,7 +243,7 @@ def get_mesh(image, depth, fnum):
     verts = pts3d.reshape(-1, 3)
     #triangles = create_triangles(image.shape[0], image.shape[1])
     #print('triangles - ok')
-    rgba = cv2.cvtColor(image[fnum], cv2.COLOR_RGB2RGBA)
+    rgba = cv2.cvtColor(image[fnum][0], cv2.COLOR_RGB2RGBA)
     colors = rgba.reshape(-1, 4)
     clrs = [[128, 128, 128, 0]]
 
@@ -317,8 +317,8 @@ with gr.Blocks(css=css) as demo:
             input_video = gr.Video(label="Input Video", format="mp4")
             input_url.change(fn=loadurl, inputs=[input_url], outputs=[input_video])
             submit = gr.Button("Submit")
-            output_frame = gr.Gallery(label="Frame", type='numpy', rows=1)
-            output_depth = gr.Gallery(label="Depth", type='numpy', rows=1)
+            output_frame = gr.Gallery(label="Frame", type='numpy', columns=8192)
+            output_depth = gr.Gallery(label="Depth", type='numpy', columns=8192)
             frame_n = gr.Slider(label="Frame #", minimum=0, maximum=0, step=1, value=0)
             render = gr.Button("Render")
         with gr.Column():