Update src/streamlit_app.py

src/streamlit_app.py

@@ -936,7 +936,7 @@ def get_fairchem_model(selected_model_name, model_path_or_name, device, selected
     if selected_model_name == "UMA Small":
         calc = FAIRChemCalculator(predictor, task_name=selected_task_type_fc)
     else:
-        calc = FAIRChemCalculator(predictor)
+        calc = FAIRChemCalculator(predictor, task_name="omol")
     return calc
 
 st.sidebar.markdown("## Input Options")
@@ -1123,14 +1123,14 @@ if atoms is not None:
                         _ = get_mace_model(MACE_MODELS["MACE MP 0a Small"], 'cpu', 'float32') # Dummy call
                         calc = get_fairchem_model(selected_model, model_path, device, selected_task_type)
                     elif model_type == "ORB":
-                        st.write("Setting up ORB calculator...")
+                        # st.write("Setting up ORB calculator...")
                         orbff = pretrained.orb_v3_conservative_inf_omat(device=device, precision=selected_default_dtype)
                         calc = ORBCalculator(orbff, device=device)
                     elif model_type == "MatterSim":
-                        st.write("Setting up MatterSim calculator...")
+                        # st.write("Setting up MatterSim calculator...")
                         calc = MatterSimCalculator(load_path=model_path, device=device)
                     elif model_type == "SEVEN_NET":
-                        st.write("Setting up SEVENNET calculator...")
+                        # st.write("Setting up SEVENNET calculator...")
                         calc = SevenNetCalculator(model=model_path, device=device)
                     calc_atoms.calc = calc