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()*255.0, depth[0].numpy()*255.0)
+    return(rgb[0], depth[0])
 
 # app.py text matter starts here
 st.title('NeRF:Neural Radiance Fields')
@@ -252,13 +252,16 @@ phi = st.slider('Enter a value for phi',min_value = 0.0,max_value = 360.0)
 
 
 color,depth = show_rendered_image(r,theta,phi)
+color = tf.keras.utils.array_to_img(color)
+depth = tf.keras.utils.array_to_img(depth)
 
-fig, ax = plt.subplots()
-im = ax.imshow(color)
-st.pyplot(fig)
+# fig, ax = plt.subplots()
+# im = ax.imshow(color)
+# st.pyplot(fig)
+
+st.image(color, caption = "Color",clamp = True, width = 200)
+st.image(depth, caption = "Depth",clamp = True, width = 200)
 
-#st.image(color.astype('int32'), caption = "Color",clamp = True)
-# st.image(depth, caption = "Depth")