fix script

script.py

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