# Base image with CUDA support | |
FROM slabstech/dhwani-server-base | |
WORKDIR /app | |
COPY . . | |
ENV HF_HOME=/data/huggingface | |
# Expose port | |
EXPOSE 7860 | |
# Start the server | |
CMD ["python", "/app/src/server/main.py", "--host", "0.0.0.0", "--port", "7860", "--config", "config_two"] |