fix script

script.py

@@ -446,32 +446,21 @@ if __name__ == "__main__":
     MODEL_PATH = "metaformer-s-224.pth"
     MODEL_NAME = "timm/vit_base_patch14_reg4_dinov2.lvd142m"
 
-    # Real submission
+    # # Real submission
     # import zipfile
 
-    with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
-        zip_ref.extractall("/tmp/data")
+    # with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
+    #     zip_ref.extractall("/tmp/data")
 
-    # metadata_file_path = "./test_preprocessed.csv"
-    # test_metadata = pd.read_csv(metadata_file_path)
-
-    # make_submission(
-    #     test_metadata=test_metadata,
-    #     model_path=MODEL_PATH,
-    #     model_name=MODEL_NAME
-    # )
+    # metadata_file_path = "./_test_preprocessed.csv"
+    # root_dir = "/tmp/data"
 
     # Test submission
-
     metadata_file_path = "../trial_submission.csv"
+    root_dir = "../data/DF_FULL"
 
-    test_metadata = pd.read_csv(metadata_file_path)
-
-    make_submission(
-        test_metadata=test_metadata,
-        model_path=MODEL_PATH,
-        model_name=MODEL_NAME,
-        images_root_path="../data/DF_FULL"
-    )
+    ##############
 
+    metadata_df = generate_embeddings(metadata_file_path, root_dir)
 
+    make_submission(metadata_df)