Dockerfile

# Utilizza l'immagine base ufficiale di Hugging Face per PyTorch con supporto GPU
FROM huggingface/transformers-pytorch-gpu:latest

# Aggiorna il sistema e installa le dipendenze di sistema
RUN apt-get update && apt-get install -y \
    python3 \
    python3-pip \
    git \
    wget \
    && rm -rf /var/lib/apt/lists/*

# Imposta un alias per python
RUN ln -s /usr/bin/python3 /usr/bin/python

# Imposta le variabili d'ambiente per il token e l'utente
ENV HUGGINGFACE_USER=Rathalos
ENV HUGGINGFACE_REPO=training_incite

# Clona il repository privato
WORKDIR /app

# Copia il file requirements.txt e installa le dipendenze Python
COPY requirements.txt .
RUN pip3 install --no-cache-dir -r requirements.txt

# Comando per eseguire lo script di training
CMD ["python", "train_model.py"]