hot dog

app.py

@@ -12,7 +12,7 @@ from demo.src.utils import render_predict_from_pose, predict_to_image
 st.set_page_config(page_title="DietNeRF Demo")
 
 def select_model():
-    obj_select = st.selectbox("Select Object to Render", ('Chair', 'Lego','Ship','hog dog'))
+    obj_select = st.selectbox("Select Object to Render", ('Chair', 'Lego','Ship','hogdog'))
     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 == 'hog dog':
+    elif obj_select == 'hogdog':
         FILE_ID = "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu"
         MODEL_DIR = 'models'
         MODEL_NAME = 'diet_nerf_hotdog'