Update script.py

script.py

@@ -17,7 +17,7 @@ def is_gpu_available():
 class PytorchWorker:
     """Run inference using ONNX runtime."""
 
-    def __init__(self, model_path: str, model_name: str, number_of_categories: int = 1605):
+    def __init__(self, model_path: str, model_name: str, number_of_categories: int = 1604):
 
         def _load_model(model_name, model_path):
 
@@ -98,7 +98,8 @@ if __name__ == "__main__":
 
     # MODEL_PATH = './e21_t152.pth'
     # MODEL_PATH = './e18_t151.pth'
-    MODEL_PATH = './e23_t141.pth'
+    # MODEL_PATH = './e23_t141.pth'
+    MODEL_PATH = './e25_t144.pth'
     MODEL_NAME = 'tf_efficientnet_b3_ns' #"tf_efficientnet_b1.ap_in1k"
 
     metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"