Ashmit24134 commited on
Commit
a6e9109
·
verified ·
1 Parent(s): 969b8fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -1,15 +1,29 @@
1
  from fastapi import FastAPI
2
  from sentence_transformers import SentenceTransformer
 
3
 
4
  app = FastAPI()
5
 
6
  model = SentenceTransformer("BAAI/bge-base-en-v1.5")
7
 
 
 
 
 
 
 
8
  @app.post("/embed")
9
  def embed(data: dict):
10
 
11
  text = data["text"]
12
 
13
- embedding = model.encode(text).tolist()
 
 
 
 
 
 
14
 
15
- return {"embedding": embedding}
 
 
1
  from fastapi import FastAPI
2
  from sentence_transformers import SentenceTransformer
3
+ import uvicorn
4
 
5
  app = FastAPI()
6
 
7
  model = SentenceTransformer("BAAI/bge-base-en-v1.5")
8
 
9
+
10
+ @app.get("/")
11
+ def root():
12
+ return {"status": "embedding api running"}
13
+
14
+
15
  @app.post("/embed")
16
  def embed(data: dict):
17
 
18
  text = data["text"]
19
 
20
+ embedding = model.encode(
21
+ text,
22
+ normalize_embeddings=True
23
+ ).tolist()
24
+
25
+ return {"embedding": embedding}
26
+
27
 
28
+ if __name__ == "__main__":
29
+ uvicorn.run(app, host="0.0.0.0", port=7860)