Changed dockerfile
Browse files- Dockerfile +4 -2
Dockerfile
CHANGED
|
@@ -11,8 +11,10 @@ RUN apt-get update && apt-get install -y \
|
|
| 11 |
COPY requirements.txt .
|
| 12 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 13 |
|
|
|
|
|
|
|
| 14 |
COPY app ./app
|
| 15 |
|
| 16 |
-
EXPOSE
|
| 17 |
|
| 18 |
-
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "
|
|
|
|
| 11 |
COPY requirements.txt .
|
| 12 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 13 |
|
| 14 |
+
RUN python -c "import torchvision.models as models; models.resnet50(weights = models.ResNet50_Weights.DEFAULT)"
|
| 15 |
+
|
| 16 |
COPY app ./app
|
| 17 |
|
| 18 |
+
EXPOSE 7860
|
| 19 |
|
| 20 |
+
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
|