UI Changes

app.py

@@ -9,10 +9,10 @@ from demo.src.models import load_trained_model
 from demo.src.utils import render_predict_from_pose, predict_to_image
 #from demo.src.config import MODEL_DIR, MODEL_NAME, FILE_ID
 
-st.set_page_config(page_title="DietNeRF Demo")
+st.set_page_config(page_title="DietNeRF")
 
 def select_model():
-    obj_select = st.selectbox("Select Object to Render", ('Chair', 'Lego','Ship','hogdog'))
+    obj_select = st.selectbox("Select an object to render", ('Chair', 'Lego','Ship','Hotdog'))
     if obj_select == 'Chair':
         FILE_ID = "17dj0pQieo94TozFv-noSBkXebduij1aM"
         MODEL_DIR = 'models'
@@ -25,7 +25,7 @@ def select_model():
         FILE_ID = "14ZeJ86ETQr8dtu6CFoxU-ifvniHKo_Dt"
         MODEL_DIR = 'models'
         MODEL_NAME = 'diet_nerf_ship'
-    elif obj_select == 'hogdog':
+    elif obj_select == 'Hotdog':
         FILE_ID = "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu"
         MODEL_DIR = 'models'
         MODEL_NAME = 'diet_nerf_hotdog'
@@ -42,7 +42,7 @@ def download_model():
     #                                     unzip=True)
     url = f'https://drive.google.com/uc?id={FILE_ID}'
     gdown.download(url, _model_path, quiet=False)
-    print(f'model downloaded from google drive: {_model_path}')
+    print(f'Model downloaded from google drive: {_model_path}')
 
 
 @st.cache(show_spinner=False, allow_output_mutation=True)
@@ -72,7 +72,9 @@ caption = "Diet-NeRF achieves SoTA few-shot learning capacity in 3D model recons
           "outperforming original NeRF!"
 st.markdown(f""" <h4> {caption} </h4> """,
             unsafe_allow_html=True)
-with st.spinner("Rendering Image (may take 2-3 mins)..."):
+st.markdown("")
+
+with st.spinner("Rendering Image, it may take 2-3 mins. So, why don't you read our report in the meantime"):
     pred_color, _ = render_predict_from_pose(state, theta, phi, radius)
     im = predict_to_image(pred_color)
     w, _ = im.size