Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# Base image with a lightweight version suitable for Hugging Face Spaces
-FROM python:3.8-slim
+# Base image with GPU support
+FROM tensorflow/tensorflow:latest-gpu
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
@@ -25,14 +25,13 @@ RUN pip install --no-cache-dir \
     Flask \
     Pillow \
     huggingface_hub \
-    tensorflow \
     tensorflow_hub \
     opencv-python
 
 # Set Hugging Face cache to a guaranteed writable directory
 ENV TRANSFORMERS_CACHE=/app/cache
 ENV HF_HOME=/app/cache
-RUN mkdir -p /app/cache && chmod -R 777 /app/cache
+RUN mkdir -p /app/cache && chown -R 1000:1000 /app/cache && chmod -R 777 /app/cache
 
 # Create directories for the models
 RUN mkdir -p /models/blip /models/clip
@@ -62,3 +61,4 @@ EXPOSE 7860
 
 # Run the Flask app
 CMD ["python", "/app/app.py"]
+