Use Keras 3

opennsfw2/_inference.py

@@ -9,7 +9,7 @@
 from PIL import Image  # type: ignore
 from tqdm import tqdm  # type: ignore
 
-from keras_core import KerasTensor, Model  # type: ignore
+from keras import KerasTensor, Model  # type: ignore
 
 from ._download import get_default_weights_path
 from ._image import preprocess_image, Preprocessing

opennsfw2/_inspection.py

@@ -5,8 +5,8 @@
 
 import numpy as np
 import tensorflow as tf  # type: ignore # pylint: disable=import-error
-from keras_core import Model  # type: ignore
-from keras_core.preprocessing.image import array_to_img  # type: ignore
+from keras import Model  # type: ignore
+from keras.preprocessing.image import array_to_img  # type: ignore
 from matplotlib import colormaps as cm
 from PIL import Image  # type: ignore
 

opennsfw2/_model.py

@@ -8,7 +8,7 @@
 import os
 from typing import Optional, Tuple
 
-from keras_core import layers, KerasTensor, Model  # type: ignore
+from keras import layers, KerasTensor, Model  # type: ignore
 
 from ._download import get_default_weights_path, download_weights_to
 

requirements.txt

@@ -1,5 +1,5 @@
 gdown>=4.2.0
-keras-core>=0.1.0
+keras>=3.0.0
 matplotlib>=3.0.0
 numpy>=1.22.0
 opencv-python>=4.0.0.0

tests/_test.py

@@ -5,7 +5,7 @@
 import unittest
 from typing import Optional, Sequence
 
-from keras_core import backend as keras_backend
+from keras import backend as keras_backend
 
 import opennsfw2 as n2