Update dockerfile

Dockerfile

@@ -1,9 +1,11 @@
-FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
+FROM nvidia/cuda:12.2.0-cudnn8-devel-ubuntu22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONDONTWRITEBYTECODE=1
+ENV TF_FORCE_GPU_ALLOW_GROWTH=true
+ENV XLA_FLAGS="--xla_gpu_enable_fast_min_max"
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -24,15 +26,16 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Set working directory
 WORKDIR /app
 
-# Copy requirements.txt
+# Copy requirements.txt but modify TensorFlow version
 COPY requirements.txt /app/
+RUN sed -i 's/tensorflow==2.18.0/tensorflow==2.15.0/' /app/requirements.txt
 
 # Install Python dependencies
 RUN pip3 install --no-cache-dir --upgrade pip setuptools wheel
 RUN pip3 install --no-cache-dir -r requirements.txt
 
-# Install OpenCV with CUDA support
-RUN pip3 install --no-cache-dir opencv-python-headless opencv-contrib-python-headless
+# Install compatible OpenCV with CUDA support
+RUN pip3 install --no-cache-dir opencv-python-headless==4.5.5.64 opencv-contrib-python-headless==4.5.5.64
 
 # Copy application code
 COPY . /app/
@@ -41,10 +44,18 @@ COPY . /app/
 RUN chmod +x /app/run_morphing.py
 RUN chmod +x /app/FILM.py
 
+# Add FILM.py fallback to CPU if CUDA fails
+RUN sed -i '/def __call__/a\        try:' /app/FILM.py && \
+    sed -i '/result = self._model/i\            # Try with GPU' /app/FILM.py && \
+    sed -i '/result = self._model/a\        except Exception as e:\n            print("GPU inference failed, falling back to CPU")\n            # Force CPU execution\n            with tf.device("/cpu:0"):\n                result = self._model(inputs, training=False)' /app/FILM.py
+
 # Set environment variables for CUDA
 ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
 ENV PATH=/usr/local/cuda/bin:${PATH}
 
+# Force TensorFlow to use compatible cuDNN version
+ENV TF_CUDNN_VERSION=8
+
 # Expose port for Streamlit
 EXPOSE 8501