Commit
•
b1502ab
1
Parent(s):
7975728
Update Dockerfile
Browse files- Dockerfile +3 -0
Dockerfile
CHANGED
|
@@ -20,5 +20,8 @@ RUN pip install gunicorn
|
|
| 20 |
RUN pip install -U flask-cors
|
| 21 |
RUN pip install opencv-python-headless==4.5.5.64
|
| 22 |
RUN pip install rembg
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
CMD ["gunicorn","-b","0.0.0.0:7860", "app:app","--timeout","950"]
|
|
|
|
| 20 |
RUN pip install -U flask-cors
|
| 21 |
RUN pip install opencv-python-headless==4.5.5.64
|
| 22 |
RUN pip install rembg
|
| 23 |
+
RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 24 |
+
RUN pip install backgroundremover
|
| 25 |
+
|
| 26 |
|
| 27 |
CMD ["gunicorn","-b","0.0.0.0:7860", "app:app","--timeout","950"]
|