Fixed Dockerfile

demo/.dockerignore

@@ -0,0 +1 @@
+flagged/

demo/Dockerfile

@@ -1,16 +1,34 @@
 # read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
 
-FROM python:3.7
+FROM tensorflow/tensorflow:2.4.2-gpu
+
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
+
+RUN apt-get update && apt-get install -y --no-install-recommends software-properties-common \
+    libsm6 libxext6 libxrender-dev curl \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN echo "**** Installing Python ****" && \
+    add-apt-repository ppa:deadsnakes/ppa &&  \
+    apt-get install -y build-essential python3.7 python3.7-dev python3-pip && \
+    curl -O https://bootstrap.pypa.io/get-pip.py && \
+    python3.7 get-pip.py && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN alias python="/usr/bin/python3.7"
 
 WORKDIR /code
 
+RUN apt-get update -y
+RUN apt install git --fix-missing -y
+
 # install dependencies
 COPY ./requirements.txt /code/requirements.txt
-RUN python3 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN python3.7 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# resolve issue with tf==2.4 and gradio versioning issue
-RUN python3 -m pip install --force-reinstall typing_extensions==4.0.0
+# resolve issue with tf==2.4 and gradio dependency collision issue
+RUN python3.7 -m pip install --force-reinstall typing_extensions==4.0.0
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -28,4 +46,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python3", "app.py"]
+CMD ["python3.7", "app.py"]