updateupdate docker

Dockerfile

@@ -36,4 +36,4 @@ COPY --chown=user . $HOME/app
 
 RUN bash $HOME/app/download_models.sh
 
-CMD python ./src/main.py
+CMD ["python", "./src/main.py"]

src/itr/router.py

@@ -32,7 +32,7 @@ def init_model(device: str):
         vlm_model = Beit3Model(device=device)
 
 
-@router.post("/retrieval/image-text")
+@router.post("/retrieval")
 async def retrieve(item: Item) -> JSONResponse:
     try:
         query_vector = vlm_model.get_embedding(input=item.query_text)