Fix build: use tensorflow-cpu, CPU torch wheels, pin numpy<2.0, fix imageio.v2

app/models/DEEP_STEGO/hide_image.py

@@ -3,7 +3,7 @@ import tempfile
 import numpy as np
 from tensorflow.keras.models import load_model
 from PIL import Image
-import imageio
+import imageio.v2 as imageio
 from app.models.DEEP_STEGO.Utils.preprocessing import normalize_batch, denormalize_batch
 
 _MODEL_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'models', 'hide.h5')

app/models/DEEP_STEGO/reveal_image.py

@@ -3,7 +3,7 @@ import tempfile
 import numpy as np
 from tensorflow.keras.models import load_model
 from PIL import Image
-import imageio
+import imageio.v2 as imageio
 from app.models.DEEP_STEGO.Utils.preprocessing import normalize_batch, denormalize_batch
 
 _MODEL_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'models', 'reveal.h5')

requirements.txt

@@ -1,10 +1,11 @@
-tensorflow
+--extra-index-url https://download.pytorch.org/whl/cpu
+tensorflow-cpu
 torch
 torchvision
 opencv-python-headless
 Pillow
 pycryptodome
 huggingface_hub
-numpy
+numpy<2.0
 imageio
 requests