Sasikanth commited on
Commit
fd0bc4a
1 Parent(s): 1a30119

updated hot dog

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from demo.src.utils import render_predict_from_pose, predict_to_image
12
  st.set_page_config(page_title="DietNeRF Demo")
13
 
14
  def select_model():
15
- obj_select = st.selectbox("Select Object to Render", ('Chair', 'Lego','Ship'))
16
  if obj_select == 'Chair':
17
  FILE_ID = "17dj0pQieo94TozFv-noSBkXebduij1aM"
18
  MODEL_DIR = 'models'
@@ -25,6 +25,10 @@ def select_model():
25
  FILE_ID = "14ZeJ86ETQr8dtu6CFoxU-ifvniHKo_Dt"
26
  MODEL_DIR = 'models'
27
  MODEL_NAME = 'diet_nerf_ship'
 
 
 
 
28
  return MODEL_DIR,MODEL_NAME,FILE_ID
29
 
30
  MODEL_DIR,MODEL_NAME,FILE_ID = select_model()
 
12
  st.set_page_config(page_title="DietNeRF Demo")
13
 
14
  def select_model():
15
+ obj_select = st.selectbox("Select Object to Render", ('Chair', 'Lego','Ship','hog dog'))
16
  if obj_select == 'Chair':
17
  FILE_ID = "17dj0pQieo94TozFv-noSBkXebduij1aM"
18
  MODEL_DIR = 'models'
 
25
  FILE_ID = "14ZeJ86ETQr8dtu6CFoxU-ifvniHKo_Dt"
26
  MODEL_DIR = 'models'
27
  MODEL_NAME = 'diet_nerf_ship'
28
+ elif obj_select == 'hog dog':
29
+ FILE_ID = "11vNlR4lMvV_AVFgVjZmKMrMWGVG7qhNu"
30
+ MODEL_DIR = 'models'
31
+ MODEL_NAME = 'diet_nerf_hotdog'
32
  return MODEL_DIR,MODEL_NAME,FILE_ID
33
 
34
  MODEL_DIR,MODEL_NAME,FILE_ID = select_model()