chore[bert]: Docker probe of concept

docker-compose.yml

@@ -2,4 +2,11 @@ services:
   ml-service:
     build: ./ml-service
     ports:
-      - 6000:6000
+      - 6000:6000
+    deploy:
+      resources:
+        limits:
+          cpus: '0.80'  # Limita a 80% de la CPU
+          memory: 6G    # Limita a 4 GB de RAM
+      devices:
+      - "/dev/nvidia0:/dev/nvidia0"

ml-service/Dockerfile

@@ -1,8 +1,19 @@
 FROM python:3.10.8-slim
 LABEL description="Sentiment classifier of tweets service"
+LABEL maintainer="RosiSoyYo"
+LABEL version="SNAPSHOT"
+
+RUN apt-get update && \
+    apt-get -y upgrade && \
+    apt-get -y autoremove && \
+    apt-get install -y
+
+RUN apt-get clean && rm -rf /var/lib/apt/lists/*
+
 WORKDIR /app
+
 COPY requirements.txt /app/requirements.txt
-RUN pip install -r requirements.txt
-COPY . /app/
-EXPOSE 6000
-CMD ["python",  "app.py"]
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+COPY . /app
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "6000"]