Spaces:
Build error
Build error
Sasikanth
commited on
Commit
•
fd0bc4a
1
Parent(s):
1a30119
updated hot dog
Browse files
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()
|