Spaces:
Running
Running
Update src/streamlit_app.py
Browse files- 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":
|