check

app.py

@@ -241,7 +241,7 @@ def show_rendered_image(r,theta,phi):
     rgb, depth = render_rgb_depth(
         nerf_loaded, rays_flat[None, ...], t_vals[None, ...], rand=False, train=False
     )
-    return(rgb[0].numpy(), depth[0].numpy())
+    return(rgb[0], depth[0])
 
 # app.py text matter starts here
 st.title('NeRF:Neural Radiance Fields')
@@ -256,7 +256,7 @@ color,depth = show_rendered_image(r,theta,phi)
 
 fig, ax = plt.subplots()
 im = ax.imshow(color)
-st.pyplot()
+st.pyplot(im)
 
 # st.image(color, caption = "Color")
 # st.image(depth, caption = "Depth")