Gaykar commited on
Commit
6ca9c83
·
verified ·
1 Parent(s): b49655b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -6,8 +6,7 @@ import uvicorn
6
 
7
  app = FastAPI(title="Medical Embedding Service")
8
 
9
- # Load model ONCE at startup
10
- print("Loading Medical RAG Model... this may take a moment.")
11
  model = SentenceTransformer("Gaykar/all-MiniLM-L6-medical-rag")
12
  print("Model loaded successfully!")
13
 
@@ -18,14 +17,14 @@ class DocumentRequest(BaseModel):
18
  texts: List[str]
19
 
20
  @app.post("/embed_query")
21
- async def embed_query(request: QueryRequest):
22
  # Uses specialized encode_query for IR tasks
23
  embedding = model.encode_query(request.text).tolist()
24
  return {"embedding": embedding}
25
 
26
  @app.post("/embed_docs")
27
- async def embed_docs(request: DocumentRequest):
28
- # Uses specialized encode_document for IR tasks
29
  embeddings = model.encode_document(request.texts).tolist()
30
  return {"embeddings": embeddings}
31
 
 
6
 
7
  app = FastAPI(title="Medical Embedding Service")
8
 
9
+
 
10
  model = SentenceTransformer("Gaykar/all-MiniLM-L6-medical-rag")
11
  print("Model loaded successfully!")
12
 
 
17
  texts: List[str]
18
 
19
  @app.post("/embed_query")
20
+ def embed_query(request: QueryRequest):
21
  # Uses specialized encode_query for IR tasks
22
  embedding = model.encode_query(request.text).tolist()
23
  return {"embedding": embedding}
24
 
25
  @app.post("/embed_docs")
26
+ def embed_docs(request: DocumentRequest):
27
+
28
  embeddings = model.encode_document(request.texts).tolist()
29
  return {"embeddings": embeddings}
30