ManasSharma07 commited on
Commit
02968f8
·
verified ·
1 Parent(s): 447a6c8

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -224,12 +224,12 @@ FAIRCHEM_MODELS = {
224
  "ESEN SM Direct All OMOL": "esen-sm-direct-all-omol"
225
  }
226
 
227
- @st.cache_resource
228
  def get_mace_model(model_path, device, selected_default_dtype):
229
  # Create a model of the specified type.
230
  return mace_mp(model=model_path, device=device, default_dtype=selected_default_dtype)
231
 
232
- @st.cache_resource
233
  def get_fairchem_model(selected_model, model_path, device, selected_task_type):
234
  predictor = pretrained_mlip.get_predict_unit(model_path, device=device)
235
  if selected_model == "UMA Small":
 
224
  "ESEN SM Direct All OMOL": "esen-sm-direct-all-omol"
225
  }
226
 
227
+ # @st.cache_resource
228
  def get_mace_model(model_path, device, selected_default_dtype):
229
  # Create a model of the specified type.
230
  return mace_mp(model=model_path, device=device, default_dtype=selected_default_dtype)
231
 
232
+ # @st.cache_resource
233
  def get_fairchem_model(selected_model, model_path, device, selected_task_type):
234
  predictor = pretrained_mlip.get_predict_unit(model_path, device=device)
235
  if selected_model == "UMA Small":