Update Dockerfile

Dockerfile

@@ -1,8 +1,19 @@
-FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+FROM python:3.10-slim
 
-WORKDIR /app
+ENV DEBIAN_FRONTEND=noninteractive
+ENV TZ=Africa/Johannesburg
+
+RUN apt-get update && apt-get install -y \
+    tzdata \
+    ffmpeg \
+    libgl1 \
+    libglib2.0-0 \
+ && ln -snf /usr/share/zoneinfo/$TZ /etc/localtime \
+ && echo $TZ > /etc/timezone \
+ && apt-get clean \
+ && rm -rf /var/lib/apt/lists/*
 
-RUN apt-get update && apt-get install -y ffmpeg git && rm -rf /var/lib/apt/lists/*
+WORKDIR /app
 
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt