Ivan Shelonik
add: remote_hub_from_pretrained
4acae3e
raw
history blame
571 Bytes
# Use the official Python base image
FROM python:3.9
# Set the working directory in the container
WORKDIR /app
# Copy the requirements.txt file and install the Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# Copy the Flask application code into the container
COPY . .
# Expose the port on which the Flask application will run
EXPOSE 5000
# Set the environment variable for Flask
ENV FLASK_APP=api_server.py
ENV TRANSFORMERS_CACHE=transformers_cache
# Run the Flask application
CMD ["flask", "run", "--host=0.0.0.0"]