Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # Base image with GPU support and TensorFlow pre-installed
-FROM tensorflow/tensorflow:2.10.0-gpu
+FROM tensorflow/tensorflow:2.15.0-gpu
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -25,6 +25,7 @@ RUN pip install --no-cache-dir \
     Flask \
     Pillow \
     huggingface_hub \
+    tensorflow==2.15.0 \
     tensorflow_hub \
     opencv-python
 
@@ -65,5 +66,4 @@ COPY app.py /app/app.py
 EXPOSE 7860
 
 # Run the Flask app
-CMD ["python", "/app/app.py"]
-
+CMD ["python", "/app/app.py"]