Spaces:
Runtime error
Runtime error
File size: 380 Bytes
1695fdf d7c7a6c 1695fdf d7c7a6c 3ccf041 d7c7a6c ff6ffb5 1695fdf ff6ffb5 1695fdf 3ccf041 d7c7a6c 3ccf041 81ef716 1695fdf d7c7a6c 1695fdf 3ccf041 81ef716 ff6ffb5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
# Use the official Hugging Face TGI Docker image as the base
FROM huggingface/transformers-pytorch-gpu:latest
RUN pip install --upgrade pip
RUN pip install transformers torch fastapi uvicorn
ENV MODEL_NAME="Hadeel11/fine-tuned-model"
ENV USE_FP16=True
COPY app.py /app/app.py
WORKDIR /app
EXPOSE 8080
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8080"]
|