Thedeezat's picture
Enable CUDA-capable Space image and GPU PyTorch wheels.
2628c88
Raw
History Blame Contribute Delete
673 Bytes
# Hugging Face Spaces (Docker SDK) — must listen on 7860
# CUDA runtime base so T4 hardware can be used when selected.
FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
RUN apt-get update && apt-get install -y --no-install-recommends \
python3 \
python3-pip \
python3-venv \
libsndfile1 \
ffmpeg \
&& rm -rf /var/lib/apt/lists/*
RUN ln -sf /usr/bin/python3 /usr/bin/python && ln -sf /usr/bin/pip3 /usr/bin/pip
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
COPY main.py .
ENV MODEL_ID=Thedeezat/ASR-Hearing-Impaired
EXPOSE 7860
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]