Update src/streamlit_app.py

src/streamlit_app.py

@@ -677,7 +677,14 @@ FAIRCHEM_MODELS = {
 }
 # Define the available ORB models
 ORB_MODELS = {
-    "V3 OMAT Conserving": "orb_v3_conservative_inf_omat",
+    "V3 OMAT Conservative (inf)": "orb-v3-conservative-inf-omat",
+    "V3 OMAT Conservative (20)": "orb-v3-conservative-20-omat",
+    "V3 OMAT Direct (inf)": "orb-v3-direct-inf-omat",
+    "V3 OMAT Direct (20)": "orb-v3-direct-20-omat",
+    "V3 MPA Conservative (inf)": "orb-v3-conservative-inf-mpa",
+    "V3 MPA Conservative (20)": "orb-v3-conservative-20-mpa",
+    "V3 MPA Direct (inf)": "orb-v3-direct-inf-mpa",
+    "V3 MPA Direct (20)": "orb-v3-direct-20-mpa",
 }
 @st.cache_resource
 def get_mace_model(model_path, device, selected_default_dtype):
@@ -917,7 +924,7 @@ if atoms is not None:
                                     results["Error"] = "Missing FairChem reference energies."
                                     calculation_possible = False
                             
-                            elif model_type == "MACE":
+                            else == "MACE":
                                 st.write("Calculating isolated atom energies with MACE...")
                                 unique_atomic_numbers = sorted(list(set(atomic_numbers)))
                                 atom_counts = {Z_unique: np.count_nonzero(atomic_numbers == Z_unique) for Z_unique in unique_atomic_numbers}