Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
-# Utilizza l'immagine base ufficiale di Hugging Face per PyTorch con supporto GPU
-FROM huggingface/transformers-pytorch-gpu:latest
+# Utilizza l'immagine base NVIDIA CUDA con supporto GPU
+FROM nvidia/cuda:12.1.0-base
 
-# Installa le dipendenze di sistema
+# Installa Python, pip e altre dipendenze di sistema
 RUN apt-get update && apt-get install -y \
     python3 \
     python3-pip \
@@ -12,9 +12,17 @@ RUN apt-get update && apt-get install -y \
 # Imposta un alias per python
 RUN ln -s /usr/bin/python3 /usr/bin/python
 
-# Copia il file requirements.txt e installa le dipendenze Python
-COPY requirements.txt .
-RUN pip3 install --no-cache-dir -r requirements.txt
+# Installa PyTorch e Transformers
+RUN pip install torch transformers datasets
+
+# Crea una directory per la cache di Hugging Face
+RUN mkdir -p /app/.cache
+
+# Imposta le variabili d'ambiente per la cache e altre configurazioni
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HUGGINGFACE_TOKEN=hf_1234567890abcdef1234567890abcdef12345678
+ENV HUGGINGFACE_USER=Rathalos
+ENV HUGGINGFACE_REPO=training_incite
 
 # Copia tutti i file dalla directory locale alla directory /app nel container
 COPY . /app