feat: dockerização do serviço

README.md

@@ -1,13 +1,10 @@
 ---
 title: Multimodal Skin Lesion Explainability
-emoji: 🩺
 colorFrom: blue
 colorTo: indigo
-sdk: gradio
-sdk_version: "4.44.1"
-python_version: "3.10"
-app_file: app.py
-pinned: false
+sdk: docker
+app_port: 7860
+startup_duration_timeout: 1h
 license: mit
 ---
 

app.py

@@ -1,28 +1,24 @@
-import sys
 import os
+import sys
 import logging
 
 logging.basicConfig(level=logging.INFO)
 
-logging.info("[app.py] starting...")
-
 ROOT = os.path.dirname(__file__)
 SRC = os.path.join(ROOT, "src")
-logging.info(f"[app.py] ROOT={ROOT}")
-logging.info(f"[app.py] SRC={SRC}")
 
 if SRC not in sys.path:
     sys.path.insert(0, SRC)
 
-logging.info("[app.py] importing main...")
 from main import demo
-logging.info("[app.py] imported main successfully")
 
 if __name__ == "__main__":
-    logging.info("[app.py] launching gradio...")
+    port = int(os.environ.get("PORT", 7860))
+
+    logging.info(f"Starting app on port {port}")
+
     demo.launch(
         server_name="0.0.0.0",
-        server_port=int(os.environ.get("PORT", 7860)),
-        show_api=False,
+        server_port=port,
         share=True
     )

dockerfile

@@ -0,0 +1,37 @@
+FROM python:3.10-slim
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=1 \
+    HF_HOME=/data/.huggingface \
+    PORT=7860
+
+RUN useradd -m -u 1000 user
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    gcc \
+    g++ \
+    git \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender1 \
+    libgomp1 \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY --chown=user requirements.txt /app/requirements.txt
+RUN pip install --upgrade pip setuptools wheel && \
+    pip install -r /app/requirements.txt
+
+COPY --chown=user . /app
+
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+EXPOSE 7860
+
+CMD ["python", "app.py"]

requirements.txt

@@ -1,6 +1,8 @@
 torch==2.4.1
 torchvision==0.19.1
 optuna==4.2.1
+requests
+gradio
 transformers==4.46.3
 matplotlib==3.7.5
 mlflow==2.17.2