Added config.json for models

app.py

@@ -1,6 +1,7 @@
 import os
 import builtins
 import math
+import json
 import streamlit as st
 import gdown
 
@@ -13,25 +14,31 @@ from demo.src.utils import render_predict_from_pose, predict_to_image
 
 st.set_page_config(page_title="DietNeRF")
 
+with open("config.json") as f:
+    cfg = json.loads(f.read())
+
 
 def select_model():
     obj_select = st.selectbox("Select a scene", ("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"
+    # 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"
+    MODEL_DIR = "models"
+    MODEL_NAME = cfg[obj_select]["DIET_NERF_MODEL_NAME"]
+    FILE_ID = cfg[obj_select]["FILE_ID"]
     return MODEL_DIR, MODEL_NAME, FILE_ID
 
 
@@ -71,7 +78,7 @@ if not os.path.isfile(model_path):
 
 model, state = fetch_model()
 pi = math.pi
-# st.sidebar.image("images/diet-nerf-logo.png", width=310)
+
 st.sidebar.markdown(
     """
 <style>