Update Dockerfile

Dockerfile

@@ -1,17 +1,10 @@
-# Use an official Python runtime as a parent image
-FROM python:3.8.9
+# Dockerfile
+FROM huggingface/transformers-pytorch-gpu
 
-# Set the working directory in the container to /app
 WORKDIR /app
 
-# Add the current directory contents into the container at /app
-ADD . /app
+COPY . .
 
-# Install any needed packages specified in requirements.txt
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install -r requirements.txt
 
-# Make port 80 available to the world outside this container
-EXPOSE 80
-
-# Run app.py when the container launches
-CMD ["python", "app.py"]
+CMD ["streamlit", "run", "app.py"]