1

data-max-hq · Jun 26, 2024 · 3ac6366 · 3ac6366
1 parent b14ff95
commit 3ac6366
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,24 @@
+# Use the DocTR image as the base image
+FROM ghcr.io/mindee/doctr:torch-py3.9.18-gpu-2024-05
+
+# Set the working directory inside the container
+WORKDIR /app
+
+# Copy the project files into the working directory
+COPY . /app
+
+# Install Python dependencies from requirements.txt
+RUN pip install -r requirements.txt
+
+# Upgrade Flask to the latest version
+RUN pip install -U Flask
+
+# Create directories for screenshots and chroma if they don't exist
+RUN mkdir -p /app/screenshots /app/chroma
+
+# Expose the Flask port
+EXPOSE 9876
+
+# Command to run the Flask app
+CMD ["flask", "--app", "app.py", "run", "--host=0.0.0.0", "--port=9876"]
+