Update Dockerfile

Dockerfile

@@ -1,3 +1,9 @@
+FROM ubuntu:18.04
+
+# Copy the package list and requirements file to the container
+COPY packages.txt requirements.txt /root/
+
+# Update the package list and install necessary packages
 RUN sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && \
     sed -i 's http://archive.ubuntu.com http://us-east-1.ec2.archive.ubuntu.com g' /etc/apt/sources.list && \
     sed -i '/security/d' /etc/apt/sources.list && \
@@ -5,18 +11,12 @@ RUN sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/so
     xargs -r -a /root/packages.txt apt-get install -y && \
     rm -rf /var/lib/apt/lists/*
 
-
-# Copy the package list and requirements file to the container
-COPY packages.txt /root/
-
 # Install Python and the packages listed in requirements.txt
 RUN apt-get update && \
-    DEBIAN_FRONTEND=noninteractive apt-get install -y python3 python3-pip && \
+    DEBIAN_FRONTEND=noninteractive apt-get install -y python3 python3-pip python3-protobuf python3-click && \
     pip3 install --no-cache-dir datasets==1.14.0 huggingface-hub==0.12.1
 
-# Install additional Python packages listed in packages.txt
-RUN xargs -r -a /root/packages.txt apt-get install -y && \
-    rm -rf /var/lib/apt/lists/*
+docker build -t my-image-name -f Dockerfile .