Update Dockerfile and main.py imports

sahi-mfg · Jan 2, 2024 · 6ab636b · 6ab636b
1 parent e91d052
commit 6ab636b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -11,4 +11,4 @@ RUN apt-get -y update && \
 COPY requirements.txt .
 RUN pip3 install -r requirements.txt
 
-CMD ["uvicorn","main:app", "--port","5001", "--host","0.0.0.0"]
+CMD ["uvicorn","app.main:app", "--port","5001", "--host","0.0.0.0"]
diff --git a/app/main.py b/app/main.py
@@ -2,7 +2,7 @@
 
 import uvicorn
 from fastapi import FastAPI, File, HTTPException, UploadFile
-from model import load_model, predict, prepare_image
+from .model import load_model, predict, prepare_image
 from PIL import Image
 from pydantic import BaseModel