Update Dockerfile

Dockerfile

@@ -12,9 +12,9 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
 # Set working directory inside the container
 WORKDIR /app
 
-# Install TensorFlow and Keras (no need for tf-keras as a separate package)
+# Install TensorFlow, Keras, and tf-keras
 # Using a single RUN command to reduce layers
-RUN pip install tensorflow==2.16.1 keras
+RUN pip install tensorflow==2.16.1 keras tf-keras
 
 # Create a non-privileged user to run the application
 # Follow best practices from Docker's user handling