Update Dockerfile

Dockerfile

@@ -11,15 +11,21 @@ RUN mkdir -p /.local && chmod -R 777 /.local && \
 # Set HOME to a writable directory
 ENV HOME=/app
 
-# Download and install the Kabyle model
-RUN curl -L -o /app/en_kab_comp.argosmodel \
-    https://huggingface.co/spaces/axxam/LibreTranslate_Kabyle/resolve/main/en_kab_comp.argosmodel && \
-    /app/venv/bin/python3 -c "import argostranslate.package; argostranslate.package.install_from_path('/app/en_kab_comp.argosmodel')"
+# Create model folder
+RUN mkdir -p /app/models
+
+# Copy model files and installer script
+COPY en_kab_comp.argosmodel /app/models/en_kab_comp.argosmodel
+COPY translate-en_kab.argosmodel /app/models/translate-en_kab.argosmodel
+COPY install_model.py /app/install_model.py
+
+# Install models
+RUN /app/venv/bin/python3 /app/install_model.py
 
 # Set environment variables
 ENV LT_HOST="0.0.0.0"
 ENV LT_PORT="7860"
-ENV LT_LOAD_ONLY="en,fr,kab_comp"
+ENV LT_LOAD_ONLY="en,fr,kab,kab_comp"
 ENV LT_SSL=True
 
 ENTRYPOINT ["/app/venv/bin/libretranslate"]