Create Dockerfile

platform/Dockerfile

@@ -1,37 +1,17 @@
-FROM python:3.11-slim-buster as prod
+FROM huggingface/transformers-pytorch-gpu:latest
 
-RUN apt-get update && apt-get install -y \
-  default-libmysqlclient-dev \
-  gcc \
-  pkg-config \
-  openjdk-11-jdk \
-  build-essential \
-  && rm -rf /var/lib/apt/lists/*
-
-RUN pip install poetry==1.4.2
-
-# Configuring poetry
-RUN poetry config virtualenvs.create false
-
-# Copying requirements of a project
-COPY pyproject.toml /app/src/
-WORKDIR /app/src
-
-# Installing requirements
-RUN poetry install --only main
-# Removing gcc
-RUN apt-get purge -y \
-  g++ \
-  gcc \
-  pkg-config \
-  && rm -rf /var/lib/apt/lists/*
+ENV DEBIAN_FRONTEND=noninteractive
 
-# Copying actual application
-COPY . /app/src/
-RUN poetry install --only main
-
-CMD ["/usr/local/bin/python", "-m", "reworkd_platform"]
-
-FROM prod as dev
-
-RUN poetry install
+RUN apt-get update && apt-get install -y \
+    git \
+    wget \
+    curl \
+    vim \
+    nano \
+    build-essential \
+    python3 \
+    python3-pip
+
+WORKDIR /app
+
+COPY . .