Update src/streamlit_app.py

src/streamlit_app.py

@@ -28,8 +28,8 @@ os.environ["STREAMLIT_WATCHER_TYPE"] = "none"
 
 # Check if running on Streamlit Cloud vs locally
 is_streamlit_cloud = os.environ.get('STREAMLIT_RUNTIME_ENV') == 'cloud'
-MAX_ATOMS_CLOUD = 50  # Maximum atoms allowed on Streamlit Cloud
-MAX_ATOMS_CLOUD_UMA = 15
+MAX_ATOMS_CLOUD = 100  # Maximum atoms allowed on Streamlit Cloud
+MAX_ATOMS_CLOUD_UMA = 50
 
 # Set page configuration
 st.set_page_config(
@@ -139,12 +139,12 @@ FAIRCHEM_MODELS = {
     "ESEN SM Direct All OMOL": "esen-sm-direct-all-omol"
 }
 
-@st.cache_resource
+#@st.cache_resource
 def get_mace_model(model_path, device, selected_default_dtype):
     # Create a model of the specified type.
     return mace_mp(model=model_path, device=device, default_dtype=selected_default_dtype)
 
-@st.cache_resource
+#@st.cache_resource
 def get_fairchem_model(selected_model, model_path, device, selected_task_type):
     predictor = pretrained_mlip.get_predict_unit(model_path, device=device)
     if selected_model == "UMA Small":