FROM ubuntu:latest

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

ENV HF_HOME="/tmp/hf"