ManasSharma07 commited on
Commit
2518e9f
·
verified ·
1 Parent(s): 383db9d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -149,7 +149,7 @@ def get_mace_model(model_path, device, selected_default_dtype):
149
  # Create a model of the specified type.
150
  return mace_mp(model=model_path, device=device, default_dtype=selected_default_dtype)
151
 
152
- #@st.cache_resource
153
  def get_fairchem_model(selected_model, model_path, device, selected_task_type):
154
  predictor = pretrained_mlip.get_predict_unit(model_path, device=device)
155
  if selected_model == "UMA Small":
 
149
  # Create a model of the specified type.
150
  return mace_mp(model=model_path, device=device, default_dtype=selected_default_dtype)
151
 
152
+ @st.cache_resource
153
  def get_fairchem_model(selected_model, model_path, device, selected_task_type):
154
  predictor = pretrained_mlip.get_predict_unit(model_path, device=device)
155
  if selected_model == "UMA Small":