Jiecong Lin
commited on
Commit
·
e697160
1
Parent(s):
0c1b085
FIX dockerfile
Browse files- Dockerfile +6 -3
- app.py +1 -10
Dockerfile
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
FROM nvidia/cuda:12.5.1-cudnn-runtime-ubuntu22.04
|
| 2 |
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
| 3 |
ENV MPLCONFIGDIR=/tmp/matplotlib
|
| 4 |
RUN apt-get update && apt-get install -y \
|
| 5 |
python3.11 \
|
|
@@ -24,6 +25,8 @@ RUN pip3 install -e .
|
|
| 24 |
# RUN pip3 install -e ".[demo]"
|
| 25 |
WORKDIR /app/segment-anything-2/checkpoints
|
| 26 |
RUN ./download_ckpts.sh
|
| 27 |
-
WORKDIR /app
|
| 28 |
-
|
| 29 |
-
|
|
|
|
|
|
|
|
|
| 1 |
FROM nvidia/cuda:12.5.1-cudnn-runtime-ubuntu22.04
|
| 2 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 3 |
+
ENV PYTHONUNBUFFERED=1
|
| 4 |
ENV MPLCONFIGDIR=/tmp/matplotlib
|
| 5 |
RUN apt-get update && apt-get install -y \
|
| 6 |
python3.11 \
|
|
|
|
| 25 |
# RUN pip3 install -e ".[demo]"
|
| 26 |
WORKDIR /app/segment-anything-2/checkpoints
|
| 27 |
RUN ./download_ckpts.sh
|
| 28 |
+
WORKDIR /app
|
| 29 |
+
EXPOSE 7860
|
| 30 |
+
ENV GRADIO_SERVER_NAME="0.0.0.0"
|
| 31 |
+
COPY app.py /app/app.py
|
| 32 |
+
CMD ["python3", "-u", "app.py"]
|
app.py
CHANGED
|
@@ -1,23 +1,14 @@
|
|
| 1 |
# import subprocess
|
| 2 |
# import re
|
| 3 |
# from typing import List, Tuple, Optional
|
| 4 |
-
|
| 5 |
-
# # Define the command to be executed
|
| 6 |
# command = ["python", "setup.py", "build_ext", "--inplace"]
|
| 7 |
-
|
| 8 |
-
# # Execute the command
|
| 9 |
# result = subprocess.run(command, capture_output=True, text=True)
|
| 10 |
-
|
| 11 |
-
# # Print the output and error (if any)
|
| 12 |
# print("Output:\n", result.stdout)
|
| 13 |
# print("Errors:\n", result.stderr)
|
| 14 |
-
|
| 15 |
-
# # Check if the command was successful
|
| 16 |
# if result.returncode == 0:
|
| 17 |
# print("Command executed successfully.")
|
| 18 |
# else:
|
| 19 |
# print("Command failed with return code:", result.returncode)
|
| 20 |
-
|
| 21 |
import gc
|
| 22 |
import math
|
| 23 |
import os
|
|
@@ -566,7 +557,7 @@ def seg_track_app():
|
|
| 566 |
)
|
| 567 |
|
| 568 |
app.queue(concurrency_count=1)
|
| 569 |
-
app.launch(debug=True, enable_queue=True, share=
|
| 570 |
|
| 571 |
if __name__ == "__main__":
|
| 572 |
seg_track_app()
|
|
|
|
| 1 |
# import subprocess
|
| 2 |
# import re
|
| 3 |
# from typing import List, Tuple, Optional
|
|
|
|
|
|
|
| 4 |
# command = ["python", "setup.py", "build_ext", "--inplace"]
|
|
|
|
|
|
|
| 5 |
# result = subprocess.run(command, capture_output=True, text=True)
|
|
|
|
|
|
|
| 6 |
# print("Output:\n", result.stdout)
|
| 7 |
# print("Errors:\n", result.stderr)
|
|
|
|
|
|
|
| 8 |
# if result.returncode == 0:
|
| 9 |
# print("Command executed successfully.")
|
| 10 |
# else:
|
| 11 |
# print("Command failed with return code:", result.returncode)
|
|
|
|
| 12 |
import gc
|
| 13 |
import math
|
| 14 |
import os
|
|
|
|
| 557 |
)
|
| 558 |
|
| 559 |
app.queue(concurrency_count=1)
|
| 560 |
+
app.launch(debug=True, enable_queue=True, share=False)
|
| 561 |
|
| 562 |
if __name__ == "__main__":
|
| 563 |
seg_track_app()
|