Spaces:
Build error
Build error
Remove persistance
Browse files
app.py
CHANGED
|
@@ -81,7 +81,7 @@ obj_select = st.selectbox("Select a Scene", SCENES_LIST, index=0)
|
|
| 81 |
DIET_NERF_MODEL_NAME, DIET_NERF_FILE_ID, NERF_MODEL_NAME, NERF_FILE_ID = select_model(obj_select)
|
| 82 |
|
| 83 |
|
| 84 |
-
@st.cache(show_spinner=False
|
| 85 |
def download_diet_nerf_model():
|
| 86 |
os.makedirs(MODEL_DIR, exist_ok=True)
|
| 87 |
diet_nerf_model_path = os.path.join(MODEL_DIR, DIET_NERF_MODEL_NAME)
|
|
@@ -97,13 +97,13 @@ def download_diet_nerf_model():
|
|
| 97 |
# print(f"Model downloaded from google drive: {nerf_model_path}")
|
| 98 |
|
| 99 |
|
| 100 |
-
@st.cache(show_spinner=False, allow_output_mutation=True
|
| 101 |
def fetch_diet_nerf_model():
|
| 102 |
model, state = load_trained_model(MODEL_DIR, DIET_NERF_MODEL_NAME)
|
| 103 |
return model, state
|
| 104 |
|
| 105 |
|
| 106 |
-
# @st.cache(show_spinner=False, allow_output_mutation=True
|
| 107 |
# def fetch_nerf_model():
|
| 108 |
# model, state = load_trained_model(MODEL_DIR, NERF_MODEL_NAME)
|
| 109 |
# return model, state
|
|
|
|
| 81 |
DIET_NERF_MODEL_NAME, DIET_NERF_FILE_ID, NERF_MODEL_NAME, NERF_FILE_ID = select_model(obj_select)
|
| 82 |
|
| 83 |
|
| 84 |
+
@st.cache(show_spinner=False)
|
| 85 |
def download_diet_nerf_model():
|
| 86 |
os.makedirs(MODEL_DIR, exist_ok=True)
|
| 87 |
diet_nerf_model_path = os.path.join(MODEL_DIR, DIET_NERF_MODEL_NAME)
|
|
|
|
| 97 |
# print(f"Model downloaded from google drive: {nerf_model_path}")
|
| 98 |
|
| 99 |
|
| 100 |
+
@st.cache(show_spinner=False, allow_output_mutation=True)
|
| 101 |
def fetch_diet_nerf_model():
|
| 102 |
model, state = load_trained_model(MODEL_DIR, DIET_NERF_MODEL_NAME)
|
| 103 |
return model, state
|
| 104 |
|
| 105 |
|
| 106 |
+
# @st.cache(show_spinner=False, allow_output_mutation=True)
|
| 107 |
# def fetch_nerf_model():
|
| 108 |
# model, state = load_trained_model(MODEL_DIR, NERF_MODEL_NAME)
|
| 109 |
# return model, state
|