update app

app.py

@@ -305,7 +305,7 @@ def reshape_image(arr):
 ##### STREAMLIT FUNCTIONS ######
 st.title('CLOOME. Bioimage database retrieval from chemical structures (and viceversa)')
 
-def main_page(top_n):
+def main_page(top_n, model_path):
     st.markdown(
     """
     Contrastive learning for self-supervised representation learning has brought a strong improvement to many application areas, such as computer vision and natural language processing. 
@@ -358,7 +358,7 @@ def molecules_from_image(top_n, model_path):
             molpath = os.path.join(datapath, "mols.hdf")
             fps_fname = save_hdf(morgan, molnames, molpath)
             mol_imgs = draw_molecules(smiles)
-            mol_features, mol_ids = main(mol_index, MODEL_PATH, model_type, mol_path=molpath, image_resolution=image_resolution)
+            mol_features, mol_ids = main(mol_index, model_path, model_type, mol_path=molpath, image_resolution=image_resolution)
             predefined_features = False
         else:
             mol_index = pd.read_csv(mol_index_file)
@@ -375,9 +375,11 @@ def molecules_from_image(top_n, model_path):
         print(img_features.shape)
         print(mol_features.shape)
 
+        top_n = int(top_n)
+
         logits = img_features @ mol_features.T
         mol_probs = (30.0 * logits).softmax(dim=-1)
-        top_probs, top_labels = mol_probs.cpu().topk(5, dim=-1)
+        top_probs, top_labels = mol_probs.cpu().topk(top_n, dim=-1)
 
         # Delete this if want to allow retrieval for multiple images
         top_probs = torch.flatten(top_probs)