Update Dockerfile

Dockerfile

@@ -1,23 +1,25 @@
-# Base image with CUDA support for PyTorch
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+# Base image
+FROM python:3.10-slim
+
+# Install system dependencies for OpenCV
+RUN apt-get update && apt-get install -y \
+    libglib2.0-0 \
+    libsm6 \
+    libxrender1 \
+    libxext6 \
+    && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
 WORKDIR /app
 
-# Copy application code
+# Copy files into container
 COPY . /app
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    python3-pip \
-    libgl1-mesa-glx \
-    && apt-get clean
-
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose port 7860
+# Expose the port used by the application
 EXPOSE 7860
 
-# Run FastAPI app with uvicorn on port 7860
+# Start the FastAPI server
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]