FROM ubuntu:latest

RUN apt update && apt install -y python3-pip && pip3 install scikit-learn datasets torch pandas

ENV HF_HOME="/tmp/hf"

COPY . .