diff --git a/embeddings/app/main.py b/embeddings/app/main.py index f0da4eb..27ea6ff 100644 --- a/embeddings/app/main.py +++ b/embeddings/app/main.py @@ -1,11 +1,11 @@ from fastapi import FastAPI from clipL14 import router as router_clip from minilm import router as router_minilm -#from finetuned import router as router_finetuned +from finetuned import router as router_finetuned app = FastAPI() app.include_router(router_clip.router, prefix="/clip") app.include_router(router_minilm.router, prefix="/minilm") -#app.include_router(router_finetuned.sbert_router, prefix="/ftbert") -#app.include_router(router_finetuned.clip_router, prefix="/ftclip") +app.include_router(router_finetuned.sbert_router, prefix="/ftbert") +app.include_router(router_finetuned.clip_router, prefix="/ftclip") diff --git a/embeddings/requirements.txt b/embeddings/requirements.txt index 56b84c3..3040f9b 100644 --- a/embeddings/requirements.txt +++ b/embeddings/requirements.txt @@ -1,4 +1,3 @@ -docarray==0.21.0 # See https://github.com/querqy/chorus/issues/169 sentence-transformers fastapi uvicorn[standard] @@ -6,4 +5,5 @@ gunicorn pydantic torch git+https://github.com/openai/CLIP.git -finetuner[full] +finetuner[full]==0.7.8 +docarray[commons]==0.21.0 diff --git a/finetuning/requirements.txt b/finetuning/requirements.txt index 82d8e9e..8ea65f3 100644 --- a/finetuning/requirements.txt +++ b/finetuning/requirements.txt @@ -1,4 +1,4 @@ -finetuner[full]=0.7.8 +finetuner[full]==0.7.8 docarray[commons]==0.21.0 click==8.1.3