Update src/streamlit_app.py

src/streamlit_app.py

@@ -999,7 +999,7 @@ if atoms is not None:
 
 st.sidebar.markdown("## Model Selection")
 if mattersim_available:
-    model_type = st.sidebar.radio("Select Model Type:", ["MACE", "FairChem", "ORB", "MatterSim"])
+    model_type = st.sidebar.radio("Select Model Type:", ["MACE", "FairChem", "ORB", "SEVEN_NET", "MatterSim"])
 else:
     model_type = st.sidebar.radio("Select Model Type:", ["MACE", "FairChem", "ORB", "SEVEN_NET"])
 
@@ -1025,8 +1025,8 @@ if model_type == "FairChem":
 if model_type == "ORB":
     selected_model = st.sidebar.selectbox("Select ORB Model:", list(ORB_MODELS.keys()))
     model_path = ORB_MODELS[selected_model]
-    # if "omat" in selected_model:
-    #     st.sidebar.warning("Using model under Academic Software License (ASL) license, see [https://github.com/gabor1/ASL](https://github.com/gabor1/ASL). To use this model you accept the terms of the license.")
+    if "omat" in selected_model:
+        st.sidebar.warning("Using model under Academic Software License (ASL) license, see [https://github.com/gabor1/ASL](https://github.com/gabor1/ASL). To use this model you accept the terms of the license.")
     # selected_default_dtype = st.sidebar.selectbox("Select Precision (default_dtype):", ['float32-high', 'float32-highest', 'float64'])
     selected_default_dtype = 'float64'
 if model_type == "MatterSim":
@@ -1034,6 +1034,8 @@ if model_type == "MatterSim":
     model_path = MATTERSIM_MODELS[selected_model]
 if model_type == "SEVEN_NET":
     selected_model = st.sidebar.selectbox("Select SEVENNET Model:", list(SEVEN_NET_MODELS.keys()))
+    if selected_model == '7net-mf-ompa':
+        selected_modal_7net = st.sidebar.selectbox("Select Modal (multi fidelity model):", ['omat', 'mpa'])
     model_path = SEVEN_NET_MODELS[selected_model]
 if atoms is not None:
     if not check_atom_limit(atoms, selected_model):
@@ -1137,7 +1139,12 @@ if atoms is not None:
                         calc = MatterSimCalculator(load_path=model_path, device=device)
                     elif model_type == "SEVEN_NET":
                         # st.write("Setting up SEVENNET calculator...")
-                        calc = SevenNetCalculator(model=model_path, device=device)
+                        if model_path=='7net-mf-omp':
+                            calc = SevenNetCalculator(model=model_path, modal=selected_modal_7net, device=device)
+                        else:
+                            calc = SevenNetCalculator(model=model_path, device=device)
+                        
+                        
                     calc_atoms.calc = calc
                     
                     if task == "Energy Calculation":