Update Dockerfile

Dockerfile

@@ -48,6 +48,11 @@ COPY --chown=hf-space:hf-space . ${HOME}/app
 
 WORKDIR ${HOME}/app
 
+# Install KenLM module
+RUN pip install --upgrade setuptools wheel
+RUN pip install https://github.com/kpu/kenlm/archive/master.zip --no-build-isolation
+
+# Install KenLM binaries
 RUN wget -O - https://kheafield.com/code/kenlm.tar.gz | tar xz && \
     mkdir kenlm/build && \
     cd kenlm/build && \
@@ -55,11 +60,9 @@ RUN wget -O - https://kheafield.com/code/kenlm.tar.gz | tar xz && \
     make -j2 && \
     cd ../..
 
-RUN mkdir ${HF_HOME} && chmod a+rwx ${HF_HOME}
-
+# Install app dependencies
 RUN pip install --no-cache-dir -r /home/hf-space/app/requirements.txt
 
-# Install KenLM module
-RUN pip install https://github.com/kpu/kenlm/archive/master.zip
+RUN mkdir ${HF_HOME} && chmod a+rwx ${HF_HOME}
 
 CMD ["python", "app.py"]