hassiahk commited on
Commit
17133ad
1 Parent(s): cf1f4d6

Remove persistance

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, persist=True)
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, persist=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, persist=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