Update Dockerfile

Dockerfile

@@ -1,15 +1,3 @@
-# Становлюємо env змінну для кешу
-# RUN mkdir -p /cache
-# RUN chmod -R 777 /cache
-
-# Становлюємо папку для result
-# RUN mkdir -p /results
-# RUN chmod -R 777 /results
-
-# Становлюємо папку для сбереження моделі
-# RUN mkdir -p /fine_tuned_model
-# RUN chmod -R 777 /fine_tuned_model
-
 # Базовий образ з Python
 FROM python:3.10-slim
 
@@ -32,7 +20,7 @@ ENV PATH="/home/user/.local/bin:$PATH"
 RUN pip install --upgrade pip
 
 # Встановлення необхідних пакетів
-RUN pip install torch transformers datasets tqdm
+RUN pip install torch time transformers datasets tqdm
 RUN pip install transformers[torch]
 
 # Скопіювати ваш Python-код у контейнер