Spaces:
Build error
Build error
Change resize factor back to 2
Browse files
app.py
CHANGED
|
@@ -18,7 +18,7 @@ with open("config.json") as f:
|
|
| 18 |
MODEL_DIR = "models"
|
| 19 |
|
| 20 |
SCENES_LIST = ["Mic", "Chair", "Lego", "Drums", "Ship", "Hotdog"]
|
| 21 |
-
random_index = random.randint(0, len(SCENES_LIST) - 1)
|
| 22 |
|
| 23 |
|
| 24 |
def select_model(obj_select):
|
|
@@ -94,7 +94,7 @@ st.markdown(
|
|
| 94 |
)
|
| 95 |
|
| 96 |
|
| 97 |
-
obj_select = st.selectbox("Select a Scene", SCENES_LIST, index=
|
| 98 |
DIET_NERF_MODEL_NAME, DIET_NERF_FILE_ID, NERF_MODEL_NAME, NERF_FILE_ID = select_model(obj_select)
|
| 99 |
|
| 100 |
|
|
@@ -150,7 +150,7 @@ with st.spinner("Rendering view..."):
|
|
| 150 |
dn_pred_color, _ = render_predict_from_pose(diet_nerf_state, theta, phi, radius)
|
| 151 |
dn_im = predict_to_image(dn_pred_color)
|
| 152 |
dn_w, _ = dn_im.size
|
| 153 |
-
dn_new_w = int(
|
| 154 |
dn_im = dn_im.resize(size=(dn_new_w, dn_new_w))
|
| 155 |
|
| 156 |
# n_pred_color, _ = render_predict_from_pose(nerf_state, theta, phi, radius)
|
|
|
|
| 18 |
MODEL_DIR = "models"
|
| 19 |
|
| 20 |
SCENES_LIST = ["Mic", "Chair", "Lego", "Drums", "Ship", "Hotdog"]
|
| 21 |
+
# random_index = random.randint(0, len(SCENES_LIST) - 1)
|
| 22 |
|
| 23 |
|
| 24 |
def select_model(obj_select):
|
|
|
|
| 94 |
)
|
| 95 |
|
| 96 |
|
| 97 |
+
obj_select = st.selectbox("Select a Scene", SCENES_LIST, index=0)
|
| 98 |
DIET_NERF_MODEL_NAME, DIET_NERF_FILE_ID, NERF_MODEL_NAME, NERF_FILE_ID = select_model(obj_select)
|
| 99 |
|
| 100 |
|
|
|
|
| 150 |
dn_pred_color, _ = render_predict_from_pose(diet_nerf_state, theta, phi, radius)
|
| 151 |
dn_im = predict_to_image(dn_pred_color)
|
| 152 |
dn_w, _ = dn_im.size
|
| 153 |
+
dn_new_w = int(2 * dn_w)
|
| 154 |
dn_im = dn_im.resize(size=(dn_new_w, dn_new_w))
|
| 155 |
|
| 156 |
# n_pred_color, _ = render_predict_from_pose(nerf_state, theta, phi, radius)
|