Update Dockerfile

Dockerfile

@@ -1,6 +1,10 @@
 # Use an official Python runtime as a parent image
 FROM python:3.9-slim-buster
 
+# Set environment variables
+ENV DISPLAY=:99
+ENV HOME=/app
+
 # Chrome browser to run the tests
 RUN apt-get update && apt-get install -y wget gnupg2 \
     && wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add - \
@@ -8,9 +12,6 @@ RUN apt-get update && apt-get install -y wget gnupg2 \
     && apt-get update && apt-get install -y google-chrome-stable \
     && apt-get clean && rm -rf /var/lib/apt/lists/* /var/cache/apt/*
 
-# Set display port to avoid crash
-ENV DISPLAY=:99
-
 # Set the working directory
 WORKDIR /app