ManasSharma07 commited on
Commit
87fb71f
·
verified ·
1 Parent(s): 94ec0e1

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -941,7 +941,7 @@ SEVEN_NET_MODELS = {
941
  }
942
  @st.cache_resource
943
  def get_mace_model(model_path, dispersion, device, selected_default_dtype):
944
- return mace_mp(model=model_path, dispersion=True, device=device, default_dtype=selected_default_dtype)
945
 
946
  @st.cache_resource
947
  def get_fairchem_model(selected_model_name, model_path_or_name, device, selected_task_type_fc): # Renamed args to avoid conflict
 
941
  }
942
  @st.cache_resource
943
  def get_mace_model(model_path, dispersion, device, selected_default_dtype):
944
+ return mace_mp(model=model_path, dispersion=dispersion, device=device, default_dtype=selected_default_dtype)
945
 
946
  @st.cache_resource
947
  def get_fairchem_model(selected_model_name, model_path_or_name, device, selected_task_type_fc): # Renamed args to avoid conflict