test with menu

app.py

@@ -26,33 +26,32 @@ from rdkit import DataStructs
 from PIL import Image
 import matplotlib.pyplot as plt
 
-
+def main_page(top_n, model_path):
+        st.markdown(
+                    """test                         """
+                                                        )
 
 #### TRYOUT MENU #####
 
 page_names_to_funcs = {
-    "Microscopy images from a molecule": images_from_molecule,
-    "Molecules from a microscopy image": molecules_from_image,
-    "About CLOOME": main_page,
+   # "Microscopy images from a molecule": images_from_molecule,
+   # "Molecules from a microscopy image": molecules_from_image,
+    "About AtomLenz": main_page,
    
 }
 
 selected_page = st.sidebar.selectbox("What would you like to retrieve?", page_names_to_funcs.keys())
 st.sidebar.markdown('')
-n_objects = st.sidebar.selectbox(
-            "How many objects would you like to retrieve?",
-            ("5", "10", "20"))
 
 
 selected_model = st.sidebar.selectbox(
-                "Select a CLOOME model to load",
-                ("CLOOME (default)", "CLOOME (CLIP imgres 320)", "CLOOME (fullres)", "CLOOME (imgres 320)"))
+                "Select a AtomLenz model to load",
+                ("AtomLenz trained on synthetic data (default)", "AtomLenz for hand-drawn images", "ChemExpert (not available yet)"))
 
 model_dict = {
-    "CLOOME (default)" : "cloome_default.pt",
-    "CLOOME (CLIP imgres 320)" : "cloome_cli_imres320.pt",
-    "CLOOME (fullres)" : "cloome_fullres.pt",
-    "CLOOME (imgres 320)" : "cloome_imres320.pt"
+    "AtomLenz trained on synthetic data (default)" : "atomlenz_default.pt",
+    "AtomLenz for hand-drawn images" : "atomlenz_handdrawn.pt",
+    "ChemExpert (not available yet)" : "atomlenz_default.pt"
 
 }