jsnfly
initial model files
9da7e2a
# Adapted from https://github.com/baaastijn/Dockerimages/tree/main/Hugginface_challenge_speech to use custom
# Transformers branch.
# Base image. Here we take one from OVHcloud with Jupyter inside and pytorch
FROM ovhcom/ai-training-pytorch:latest
# Install git, audio loader ang git lfs
RUN apt-get update && \
apt-get install -y git && \
apt-get install -y libsndfile1-dev sox && \
apt-get install -y git-lfs && \
git lfs install
# Install required python libraries. We install transformers from source to get latest version
RUN pip install --upgrade pip && \
pip install git+https://github.com/jsnfly/transformers.git@speech-challenge-experiments && \
pip install git+https://github.com/huggingface/datasets && \
pip install torchaudio librosa jiwer && \
pip install pandas numpy nano gradio
# Create a HOME dedicated to the OVHcloud user (42420:42420)
RUN chown -R 42420:42420 /workspace
ENV HOME /workspace
WORKDIR /workspace
# Copy a folder of example notebooks into another folder in remote workspace
# COPY notebooks /workspace/