UI Fixes

app.py

@@ -132,9 +132,8 @@ with st.spinner("Rendering view..."):
         # n_im = n_im.resize(size=(n_new_w, n_new_w))
 
         # diet_nerf_col, nerf_col = st.beta_columns([1, 1])
-        diet_nerf_col = st.beta_columns([1])
-        diet_nerf_col.markdown(f"""<h4 style='text-align: center'>Rendered view for {obj_select}</h4>""", unsafe_allow_html=True)
-        diet_nerf_col.image(dn_im, use_column_width=True)
+        st.markdown(f"""<h4 style='text-align: center'>Rendered view for {obj_select}</h4>""", unsafe_allow_html=True)
+        st.image(dn_im, use_column_width=True)
 
         # nerf_col.markdown("""<h4 style='text-align: center'>NeRF</h4>""", unsafe_allow_html=True)
         # nerf_col.image(n_im, use_column_width=True)