set DOWNSAMPLE=4 and then resize a bigger image

app.py

@@ -71,5 +71,7 @@ st.markdown(f""" <h4> {caption} </h4> """,
 with st.spinner("Rendering Image (may take 2-3 mins)..."):
     pred_color, _ = render_predict_from_pose(state, theta, phi, radius)
     im = predict_to_image(pred_color)
-    
+    w, _ = im.size
+    new_w = int(2*w)
+    im = im.resize(size=(new_w, new_w))
     st.image(im, use_column_width=True)

demo/src/config.py

@@ -41,4 +41,4 @@ class NerfConfig:
     FOCAL = 555.5555155968841
     # reduce CHUNK if OOM
     CHUNK = 4096
-    DOWNSAMPLE = 2
+    DOWNSAMPLE = 4

demo/src/utils.py

@@ -23,7 +23,7 @@ def render_predict_from_pose(state, theta, phi, radius):
     return pred_color, pred_disp
 
 
-def predict_to_image(pred_out):
+def predict_to_image(pred_out) -> Image:
     image_arr = np.array(np.clip(pred_out, 0., 1.) * 255.).astype(np.uint8)
     return Image.fromarray(image_arr)