checking model

app.py

@@ -48,6 +48,7 @@ def atomlenz(modelfile):
     dir_list = [os.path.join(experiment_path_atoms,f) for f in dir_list]
     dir_list.sort(key=os.path.getctime, reverse=True)
     checkpoint_file_atoms = [f for f in dir_list if "ckpt" in f][0]
+    checkpoint_file_atoms=os.path.join(experiment_path_atoms,modelfile)
     model_atom = model_cls.load_from_checkpoint(checkpoint_file_atoms)
     model_atom.model.roi_heads.score_thresh = 0.65
     experiment_path_bonds = "./models/bonds_model/"
@@ -250,9 +251,9 @@ selected_model = st.sidebar.selectbox(
                 ("AtomLenz trained on synthetic data (default)", "AtomLenz for hand-drawn images", "ChemExpert (not available yet)"))
 
 model_dict = {
-    "AtomLenz trained on synthetic data (default)" : "synthetic",
-    "AtomLenz for hand-drawn images" : "real",
-    "ChemExpert (not available yet)" : "synthetic"
+    "AtomLenz trained on synthetic data (default)" : "synthetic.ckpt",
+    "AtomLenz for hand-drawn images" : "real.ckpt",
+    "ChemExpert (not available yet)" : "synthetic.ckpt"
 
 }