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