TingTing1999
/

submit

Model card Files Files and versions

TingTing1999 commited on May 24, 2024

Commit

cb29204

·

verified ·

1 Parent(s): 6e8e572

Update script.py

Files changed (1) hide show

script.py +2 -1

script.py CHANGED Viewed

@@ -96,7 +96,8 @@ if __name__ == "__main__":
     with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
         zip_ref.extractall("/tmp/data")
-    MODEL_PATH = './e21_t152.pth'
     MODEL_NAME = 'tf_efficientnet_b3_ns' #"tf_efficientnet_b1.ap_in1k"
     metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"

     with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
         zip_ref.extractall("/tmp/data")
+    # MODEL_PATH = './e21_t152.pth'
+    MODEL_PATH = './e18_t151.pth'
     MODEL_NAME = 'tf_efficientnet_b3_ns' #"tf_efficientnet_b1.ap_in1k"
     metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"