Spaces:
Running
Running
# Use an official PyTorch runtime as parent image | |
FROM pytorch/pytorch:1.10.0-cuda11.1-cudnn8-runtime | |
# Set the working directory in the container | |
WORKDIR /app | |
# Copy the current directory contents into the container at /app | |
COPY . /app | |
# Install any needed packages specified in requirements.txt | |
RUN pip install --no-cache-dir -r requirements.txt | |
# Define environment variable | |
ENV NAME World | |
# Run app.py when the container launches | |
CMD ["python", "app.py"] | |