-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDockerfile
More file actions
27 lines (21 loc) · 768 Bytes
/
Dockerfile
File metadata and controls
27 lines (21 loc) · 768 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
# Use a base image with Python
FROM pytorch/pytorch:latest
# Install git and other system dependencies
RUN apt-get update && apt-get install -y \
git \
fontconfig \
&& rm -rf /var/lib/apt/lists/*
RUN pip install git+https://github.com/lauracabayol/TEMPS.git
# Set the working directory
WORKDIR /app
# Copy the application file(s)
COPY app.py .
# Create models directory and copy model files
RUN mkdir -p data/models
# Copy model files - make sure these files exist in your repository
COPY data/models/modelF_DA.pt data/models/
COPY data/models/modelZ_DA.pt data/models/
# Expose the port the app runs on (if needed)
EXPOSE 7860
# Updated command with correct argument names
CMD ["python", "app.py", "--port", "7860", "--server-address", "127.0.0.1"]