dockerfile big update cuda

Dockerfile

@@ -1,11 +1,30 @@
-FROM python:3.9
+FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu20.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt update && \
+    apt install -y bash \
+                   build-essential \
+                   git \
+                   git-lfs \
+                   curl \
+                   ca-certificates \
+                   libsndfile1-dev \
+                   libgl1 \
+                   python3.8 \
+                   python3-pip \
+                   python3.8-venv && \
+    rm -rf /var/lib/apt/lists
+
+RUN python3 -m venv /opt/venv
 WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN python3 -m pip install --no-cache-dir --upgrade pip && \
+    python3 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
 	PATH=/home/user/.local/bin:$PATH
 WORKDIR $HOME/server
 COPY --chown=user . $HOME/server
-CMD ["uvicorn", "api.main:api", "--host", "0.0.0.0", "--port", "7860"]
+
+CMD ["uvicorn", "api.main:api", "--host", "0.0.0.0", "--port", "7860"]

api/main.py

@@ -6,7 +6,7 @@ api = __main.api()
 def get_text_from_url(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
-        if data:
+        if data and 'url' in data:
             __response['texto']=__main.obtener_texto(from_url=data.get('url'))
         else:
             raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
@@ -21,7 +21,7 @@ def get_text_from_url(data:dict) -> dict:
 def get_text_from_pdf(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
-        if data:
+        if data and 'pdf' in data:
             __response['texto']=__main.obtener_texto(from_pdf=data.get('pdf'))
         else:
             raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
@@ -36,7 +36,7 @@ def get_text_from_pdf(data:dict) -> dict:
 def get_blocks(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
-        if data:
+        if data and 'texto' in data and 'size' in data:
             __response['original']=data.get('texto')
             __response['bloques']=__main.generar_bloques(texto=data.get('texto'),
                                                          size=data.get('size'))
@@ -53,7 +53,7 @@ def get_blocks(data:dict) -> dict:
 def get_traduccion(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
-        if data:
+        if data and 'texto' in data and 'idioma' in data:
             __response['original']= data.get('texto')
             __response['traduccion']= __main.traducir(texto=data.get('texto'),
                                                       idioma=data.get('idioma'))
@@ -70,7 +70,7 @@ def get_traduccion(data:dict) -> dict:
 def get_resumen(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
-        if data:
+        if data and 'texto' in data:
             __response['original']= data.get('texto')
             __response['resumen']= __main.resumir(texto=data.get('texto'))
         else:
@@ -86,7 +86,7 @@ def get_resumen(data:dict) -> dict:
 def get_text2img(data:dict) -> dict:
     __response=dict({"request_data":data})
     try:
-        if data:
+        if data and 'texto' in data and 'modelo' in data:
             __response['original']= data.get('texto')
             __response['image']= __main.text_to_img(texto=data.get('texto'), 
                                                     model=data.get('modelo'))

requirements.txt

@@ -1,3 +1,19 @@
+torch
+torchvision 
+torchaudio
+invisible_watermark
+datasets
+hf-doc-builder
+huggingface-hub
+Jinja2
+librosa
+numpy
+scipy
+tensorboard
+omegaconf
+pytorch-lightning
+xformers
+
 fastapi
 pydantic
 uvicorn
@@ -9,3 +25,5 @@ transformers[sentencepiece]
 diffusers
 diffusers[torch]
 diffusers[flax]
+accelerate
+safetensors