# 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"]