| version: '3.8' | |
| services: | |
| unified-model-app: | |
| build: | |
| context: . | |
| dockerfile: Dockerfile | |
| container_name: pytorch-unified-model-comparison | |
| ports: | |
| - "7860:7860" | |
| environment: | |
| PYTHONUNBUFFERED: 1 | |
| GRADIO_SERVER_NAME: 0.0.0.0 | |
| GRADIO_SERVER_PORT: 7860 | |
| volumes: | |
| # Optional: Mount models directory if you want to update models without rebuilding | |
| - ./models:/app/models | |
| restart: unless-stopped | |
| # Optional: Increase memory limit if needed | |
| # deploy: | |
| # resources: | |
| # limits: | |
| # memory: 8G | |
| # reservations: | |
| # memory: 4G | |