Update Dockerfile

Dockerfile

@@ -1,21 +1,27 @@
-# Use a CPU-based Python image compatible with Hugging Face
-FROM python:3.8-slim
+# Use TensorFlow GPU image
+FROM tensorflow/tensorflow:2.10.0-gpu
 
-# Set the working directory
+# ✅ Manually install TensorFlow CPU version
+RUN pip install --no-cache-dir tensorflow==2.10.0
+
+# Set working directory
 WORKDIR /app
 
-# Copy requirements and install dependencies
+# Ensure /tmp is writable
+RUN chmod -R 777 /tmp
+
+# Copy requirements and install
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# ✅ Manually install TensorFlow CPU version
-RUN pip install --no-cache-dir tensorflow==2.10.0
-
-# Copy the FastAPI application
+# Copy app source
 COPY app/ ./app/
 
-# Expose the port (Hugging Face expects 7860)
+# Ensure app directory is readable
+RUN chmod -R 755 /app
+
+# Expose port (Hugging Face Spaces expects 7860)
 EXPOSE 7860
 
-# Run the FastAPI app
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Run FastAPI app
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]