mlabonne commited on
Commit
0ca983c
·
1 Parent(s): a615f36

Run ColBERT Space with Docker

Browse files
Files changed (3) hide show
  1. Dockerfile +22 -0
  2. README.md +2 -3
  3. app.py +1 -1
Dockerfile ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.11-slim-bookworm
2
+
3
+ ENV PYTHONUNBUFFERED=1 \
4
+ PIP_NO_CACHE_DIR=1 \
5
+ HF_HOME=/home/user/.cache/huggingface
6
+
7
+ RUN useradd -m -u 1000 user
8
+
9
+ WORKDIR /app
10
+ COPY --chown=1000:1000 requirements.txt .
11
+
12
+ RUN pip install --upgrade pip \
13
+ && pip install --index-url https://download.pytorch.org/whl/cpu "torch==2.8.0+cpu" \
14
+ && pip install -r requirements.txt \
15
+ && pip install "gradio[oauth,mcp]==5.49.1" "uvicorn>=0.14.0" "websockets>=10.4" spaces
16
+
17
+ COPY --chown=1000:1000 . .
18
+
19
+ USER 1000
20
+ EXPOSE 7860
21
+
22
+ CMD ["python", "app.py"]
README.md CHANGED
@@ -3,9 +3,8 @@ title: LFM2 ColBERT
3
  emoji: 🏆
4
  colorFrom: purple
5
  colorTo: pink
6
- sdk: gradio
7
- sdk_version: 5.49.1
8
- app_file: app.py
9
  pinned: false
10
  short_description: One Model to Embed Them All
11
  ---
 
3
  emoji: 🏆
4
  colorFrom: purple
5
  colorTo: pink
6
+ sdk: docker
7
+ app_port: 7860
 
8
  pinned: false
9
  short_description: One Model to Embed Them All
10
  ---
app.py CHANGED
@@ -258,4 +258,4 @@ with gr.Blocks(title="Cross-Lingual Retrieval Demo", theme=gr.themes.Soft(primar
258
 
259
 
260
  if __name__ == "__main__":
261
- demo.launch()
 
258
 
259
 
260
  if __name__ == "__main__":
261
+ demo.launch(server_name="0.0.0.0", server_port=7860)