update embeddings from resnet model

app.py

@@ -329,7 +329,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist') as demo:
                 
             with gr.Column():
                 model_name = gr.Dropdown(
-                    ["Mummified 170", "Rock 170","Fossils 142"],#"Fossils BEiT" removed
+                    ["Fossils 142"],#"Mummified 170", "Rock 170","Fossils BEiT" removed
                     multiselect=False,
                     value="Fossils 142", # default option
                     label="Model",

closest_sample.py

@@ -18,9 +18,9 @@ if not os.path.exists('dataset'):
   snapshot_download(repo_id=REPO_ID, token=token,repo_type='dataset',local_dir='dataset')
 
 
-fossils_pd= pd.read_csv('fossils_paths.csv')
+fossils_pd= pd.read_csv('all_fossils_filtered_100.csv')
 
-def pca_distance(pca,sample,embedding,top_k): 
+def pca_distance(pca,sample,embedding,top_k):
     """
     Args:
         pca:fitted PCA model
@@ -62,7 +62,7 @@ def get_images(embedding,model_name):
     elif model_name in ['Fossils 142']:
         pca_fossils = pk.load(open('pca_fossils_142_resnet.pkl','rb'))
         pca_leaves = pk.load(open('pca_leaves_142_resnet.pkl','rb'))
-        embedding_fossils = np.load('dataset/embedding_leaves_142_finer.npy')
+        embedding_fossils = np.load('dataset/embedding_fossils_142_finer.npy')
         #embedding_leaves = np.load('embedding_leaves.npy')
     else:
         print(f'{model_name} not recognized')
@@ -110,7 +110,7 @@ def get_diagram(embedding,top_k,model_name):
     elif model_name in ['Fossils 142']:
         pca_fossils = pk.load(open('pca_fossils_142_resnet.pkl','rb'))
         pca_leaves = pk.load(open('pca_leaves_142_resnet.pkl','rb'))
-        embedding_fossils = np.load('dataset/embedding_leaves_142_finer.npy')
+        embedding_fossils = np.load('dataset/embedding_fossils_142_finer.npy')
         #embedding_leaves = np.load('embedding_leaves.npy')
     else:
         print(f'{model_name} not recognized')

embedding_fossils_142_finer.npy

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:284a9154c60e41c9d912c0f2e335ea01c6741e88bc779cb4c9e09e8ec05baf29
+size 28172416