muryshev commited on
Commit
9c43066
·
1 Parent(s): babc381

Revert "update"

Browse files

This reverts commit e55c5d20d93c533609404388ae0a69b1e4acacf4.

# Conflicts:
# Dockerfile

Files changed (1) hide show
  1. Dockerfile +11 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Use an official Python runtime as a parent image
2
- FROM python:3.10-slim-bullseye
3
 
4
  # Set Python to use unbuffered mode
5
  ENV PYTHONUNBUFFERED=1
@@ -28,10 +28,19 @@ USER myuser
28
  # Copy the current directory contents into the container at /var/www
29
  COPY . /var/www
30
 
 
31
  RUN pip install --user -r requirements.txt
 
 
32
 
33
  # Expose the port
34
  EXPOSE 7860
35
 
 
36
 
37
- CMD ["python", "app.py"]
 
 
 
 
 
 
1
  # Use an official Python runtime as a parent image
2
+ FROM nvidia/cuda:12.3.1-runtime-ubuntu22.04
3
 
4
  # Set Python to use unbuffered mode
5
  ENV PYTHONUNBUFFERED=1
 
28
  # Copy the current directory contents into the container at /var/www
29
  COPY . /var/www
30
 
31
+ # Install Python dependencies
32
  RUN pip install --user -r requirements.txt
33
+ # RUN pip install --user torch==1.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
34
+ # RUN pip install --user ../whl/semantic_search_multistep_data/whl/torch-1.12.1+cu113-cp310-cp310-linux_x86_64.whl
35
 
36
  # Expose the port
37
  EXPOSE 7860
38
 
39
+ RUN mkdir /var/www/logs
40
 
41
+ # Set environment variables
42
+ ENV MODEL_PATH="BAAI/bge-m3" \
43
+ DEVICE="gpu"
44
+
45
+ # Run fastapi_app.py when the container launches
46
+ CMD python3 -m uvicorn search_api:app --host=0.0.0.0 --port=7860