Update script.py

script.py

@@ -67,13 +67,13 @@ def make_submission(test_metadata, model_paths, model_name, output_csv_path="./s
         image_path = os.path.join(images_root_path, row.filename)
 
         test_image = Image.open(image_path).convert("RGB")
-        flipped_image = test_image.transpose(Image.FLIP_LEFT_RIGHT)
+        # flipped_image = test_image.transpose(Image.FLIP_LEFT_RIGHT)
 
         result_logits = []
 
         for model in models:
             result_logits += model.predict_image(test_image)
-            result_logits += model.predict_image(flipped_image)
+            # result_logits += model.predict_image(flipped_image)
 
         logits = np.average(np.array(result_logits), 0)
 
@@ -93,8 +93,8 @@ if __name__ == "__main__":
         zip_ref.extractall("/tmp/data")
 
     # MODEL_PATH = "pytorch_model.bin"
-    MODEL_PATH = ["2405_PSCmetric_real3_best_accuracy.pth",
-                  "2405_cls_boost_best_accuracy.pth"
+    MODEL_PATH = ["__best_accuracy.pth",
+                  # "2405_cls_boost_best_accuracy.pth"
                  ]
     # MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
     MODEL_NAME = "swinv2_tiny_window16_256.ms_in1k"