diff --git a/README.md b/README.md index 67983e83..0ac0cd47 100644 --- a/README.md +++ b/README.md @@ -242,7 +242,7 @@ to learn more! ## Licensing -Copyright 2024 Protect AI +Copyright 2026 Protect AI Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/modelscan/settings.py b/modelscan/settings.py index 56b3a796..de25d78b 100644 --- a/modelscan/settings.py +++ b/modelscan/settings.py @@ -75,12 +75,12 @@ class SupportedModelFormats: "middlewares": { "modelscan.middlewares.FormatViaExtensionMiddleware": { "formats": { - SupportedModelFormats.TENSORFLOW: [".pb"], - SupportedModelFormats.KERAS_H5: [".h5"], - SupportedModelFormats.KERAS: [".keras"], - SupportedModelFormats.NUMPY: [".npy"], - SupportedModelFormats.PYTORCH: [".bin", ".pt", ".pth", ".ckpt"], - SupportedModelFormats.PICKLE: [ + SupportedModelFormats.TENSORFLOW.value: [".pb"], + SupportedModelFormats.KERAS_H5.value: [".h5"], + SupportedModelFormats.KERAS.value: [".keras"], + SupportedModelFormats.NUMPY.value: [".npy"], + SupportedModelFormats.PYTORCH.value: [".bin", ".pt", ".pth", ".ckpt"], + SupportedModelFormats.PICKLE.value: [ ".pkl", ".pickle", ".joblib",