Code formatting

app.py

@@ -1,13 +1,13 @@
-import os
-import builtins
-import math
 import json
+import math
+import os
 
-import streamlit as st
 import gdown
+import streamlit as st
+
 
 from demo.src.models import load_trained_model
-from demo.src.utils import render_predict_from_pose, predict_to_image
+from demo.src.utils import predict_to_image, render_predict_from_pose
 
 st.set_page_config(page_title="DietNeRF")
 
@@ -55,13 +55,28 @@ st.sidebar.markdown(
 )
 st.sidebar.header("SELECT YOUR VIEW DIRECTION")
 theta = st.sidebar.slider(
-    "Rotation (Left to Right)", min_value=-pi, max_value=pi, step=0.5, value=0.0, help="Rotational angle in Horizontal direction (Theta)"
+    "Rotation (Left to Right)",
+    min_value=-pi,
+    max_value=pi,
+    step=0.5,
+    value=0.0,
+    help="Rotational angle in Horizontal direction (Theta)",
 )
 phi = st.sidebar.slider(
-    "Rotation (Down to Up)", min_value=0.0, max_value=0.5 * pi, step=0.1, value=1.0, help="Rotational angle in Vertical direction (Phi)"
+    "Rotation (Down to Up)",
+    min_value=0.0,
+    max_value=0.5 * pi,
+    step=0.1,
+    value=1.0,
+    help="Rotational angle in Vertical direction (Phi)",
 )
 radius = st.sidebar.slider(
-    "Distance (Close to Far)", min_value=2.0, max_value=6.0, step=1.0, value=3.0, help="Distance between object and the viewer (Radius)"
+    "Distance (Close to Far)",
+    min_value=2.0,
+    max_value=6.0,
+    step=1.0,
+    value=3.0,
+    help="Distance between object and the viewer (Radius)",
 )
 caption = (
     "`DietNeRF` achieves state-of-the-art few-shot learning capacity in 3D model reconstruction. "
@@ -90,6 +105,7 @@ def download_diet_nerf_model():
     print(f"Model downloaded from google drive: {diet_nerf_model_path}")
 
 
+# @st.cache(show_spinner=False)
 # def download_nerf_model():
 #     nerf_model_path = os.path.join(MODEL_DIR, NERF_MODEL_NAME)
 #     url = f"https://drive.google.com/uc?id={NERF_FILE_ID}"