unfinity commited on
Commit
ad29a66
1 Parent(s): fa822b0
Files changed (5) hide show
  1. Dockerfile +3 -19
  2. Not_Dockerfile +23 -0
  3. README.md +0 -1
  4. fastapi_dummy +1 -0
  5. main.py +7 -0
Dockerfile CHANGED
@@ -1,23 +1,7 @@
1
  FROM nvcr.io/nvidia/pytorch:23.12-py3
2
 
3
- # YOLOv8
4
- RUN pip install ultralytics
5
- RUN pip install opencv-python==4.6.0.66
6
- RUN pip install Pillow==10.3.0
7
 
8
- # FastAPI
9
- RUN pip install fastapi uvicorn
10
 
11
- # Streamlit
12
- # RUN pip install streamlit
13
-
14
- # Fonts
15
- RUN apt update && apt install fonts-dejavu -y
16
-
17
- # Download model weights
18
- WORKDIR /app
19
- RUN wget https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-pose.pt -O /app/yolov8l-pose.pt
20
-
21
- COPY . /app
22
-
23
- CMD ["python", "start.py"]
 
1
  FROM nvcr.io/nvidia/pytorch:23.12-py3
2
 
3
+ WORKDIR /code
 
 
 
4
 
5
+ COPY . .
 
6
 
7
+ CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
 
 
 
 
 
 
 
 
 
 
 
 
Not_Dockerfile ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvcr.io/nvidia/pytorch:23.12-py3
2
+
3
+ # YOLOv8
4
+ RUN pip install ultralytics
5
+ RUN pip install opencv-python==4.6.0.66
6
+ RUN pip install Pillow==10.3.0
7
+
8
+ # FastAPI
9
+ RUN pip install fastapi uvicorn
10
+
11
+ # Streamlit
12
+ # RUN pip install streamlit
13
+
14
+ # Fonts
15
+ RUN apt update && apt install fonts-dejavu -y
16
+
17
+ # Download model weights
18
+ WORKDIR /app
19
+ RUN wget https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-pose.pt -O /app/yolov8l-pose.pt
20
+
21
+ COPY . /app
22
+
23
+ CMD ["python", "start.py"]
README.md CHANGED
@@ -4,7 +4,6 @@ emoji: 📚
4
  colorFrom: purple
5
  colorTo: pink
6
  sdk: docker
7
- app_port: 8501
8
  pinned: false
9
  license: agpl-3.0
10
  ---
 
4
  colorFrom: purple
5
  colorTo: pink
6
  sdk: docker
 
7
  pinned: false
8
  license: agpl-3.0
9
  ---
fastapi_dummy ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit 62c7f50be22e00fc74edbfc94c16a90e0f3dda1c
main.py ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+
3
+ app = FastAPI()
4
+
5
+ @app.get("/")
6
+ def read_root():
7
+ return {"Hello": "World!"}