UI Changes

app.py

@@ -2,6 +2,7 @@ import os
 import builtins
 import math
 import json
+
 import streamlit as st
 import gdown
 
@@ -19,23 +20,7 @@ with open("config.json") as f:
 
 
 def select_model():
-    obj_select = st.selectbox("Select a scene", ("Mic", "Chair", "Lego", "Ship", "Hotdog"))
-    # if obj_select == "Chair":
-    #     FILE_ID = "17dj0pQieo94TozFv-noSBkXebduij1aM"
-    #     MODEL_DIR = "models"
-    #     MODEL_NAME = "diet_nerf_chair"
-    # elif obj_select == "Lego":
-    #     FILE_ID = "1D9I-qIVMPaxuCHfUWPWMHaoLYtAmCjwI"
-    #     MODEL_DIR = "models"
-    #     MODEL_NAME = "diet_nerf_lego"
-    # elif obj_select == "Ship":
-    #     FILE_ID = "14ZeJ86ETQr8dtu6CFoxU-ifvniHKo_Dt"
-    #     MODEL_DIR = "models"
-    #     MODEL_NAME = "diet_nerf_ship"
-    # elif obj_select == "Hotdog":
-    #     FILE_ID = "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu"
-    #     MODEL_DIR = "models"
-    #     MODEL_NAME = "diet_nerf_hotdog"
+    obj_select = st.selectbox("Select a Scene", ("Mic", "Chair", "Lego", "Ship", "Hotdog"))
     MODEL_DIR = "models"
     MODEL_NAME = cfg[obj_select]["DIET_NERF_MODEL_NAME"]
     FILE_ID = cfg[obj_select]["FILE_ID"]
@@ -58,9 +43,6 @@ MODEL_DIR, MODEL_NAME, FILE_ID = select_model()
 def download_model():
     os.makedirs(MODEL_DIR, exist_ok=True)
     _model_path = os.path.join(MODEL_DIR, MODEL_NAME)
-    # gdd.download_file_from_google_drive(file_id=FILE_ID,
-    #                                     dest_path=_model_path,
-    #                                     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}")
@@ -87,7 +69,7 @@ st.sidebar.markdown(
 }
 </style>
 <p class="aligncenter">
-    <img src="https://user-images.githubusercontent.com/77657524/126361638-4aad58e8-4efb-4fc5-bf78-f53d03799e1e.png" width="420" height="400"/>
+    <img src="https://user-images.githubusercontent.com/77657524/126361638-4aad58e8-4efb-4fc5-bf78-f53d03799e1e.png" width="430" height="400"/>
 </p>
 """,
     unsafe_allow_html=True,
@@ -120,7 +102,9 @@ with st.spinner("Rendering View..."):
         w, _ = im.size
         new_w = int(2 * w)
         im = im.resize(size=(new_w, new_w))
-        st.markdown(
-            """<h4 style='text-align: center'>DietNerF</h4>""", unsafe_allow_html=True
-        )
+
+        # diet_nerf_col = st.beta_columns(1)
+        # diet_nerf_col.markdown(
+        #     """<h4 style='text-align: center'>DietNerF</h4>""", unsafe_allow_html=True
+        # )
         st.image(im, use_column_width="auto")