Update Dockerfile

Dockerfile

@@ -1,25 +1,23 @@
 # Base image
-FROM python:3.10-slim
+FROM huggingface/transformers-pytorch-gpu:4.33.2
 
-
-# Install system dependencies for OpenCV
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     libglib2.0-0 \
-    libsm6 \
-    libxrender1 \
-    libxext6 \
+    libgl1-mesa-glx \
+    libgthread-2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set working directory
+# Set the working directory
 WORKDIR /app
 
-# Copy files into container
+# Copy the project files
 COPY . /app
 
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose the port used by the application
+# Expose the port (matching your app configuration)
 EXPOSE 7860
 
 # Start the FastAPI server