File size: 355 Bytes
e98d6df 8f0a43c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# Use an official PyTorch image with CUDA support as a base image
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
# Install additional Python dependencies
RUN pip install transformers==4.31.0 gradio==3.40.1
# Expose the default port for Gradio
EXPOSE 7860
# Copy your app files
COPY . /app
WORKDIR /app
# Run the Gradio app
CMD ["python", "app.py"] |