Spaces:
Running
Running
File size: 836 Bytes
060adf9 f6e220a 060adf9 f6e220a 060adf9 f6e220a ada2814 f6e220a ada2814 060adf9 ada2814 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# Use Python base image
FROM python:3.9
# Set working directory
WORKDIR /app
# Copy files
COPY requirements.txt .
COPY main.py .
# Install dependencies
RUN pip install --no-cache-dir -r requirements.txt
# Set cache directory (Use HF_HOME instead of TRANSFORMERS_CACHE)
ENV HF_HOME="/tmp/huggingface"
# Ensure the directory exists
RUN mkdir -p $HF_HOME
# Pre-download the correct models (Multilingual + English sentiment analysis)
RUN python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='tabularisai/multilingual-sentiment-analysis')"
RUN python -c "from transformers import pipeline; pipeline('sentiment-analysis', model='siebert/sentiment-roberta-large-english')"
# Expose port for FastAPI
EXPOSE 7860
# Run FastAPI server
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
|