Shubam407 commited on
Commit
4c4607d
·
verified ·
1 Parent(s): 49198df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -27,18 +27,26 @@ app = FastAPI(
27
  class TextRequest(BaseModel):
28
  text: str
29
 
 
 
 
 
30
  @app.get("/health")
31
  def health():
32
  return {"status": "ok"}
33
 
34
  @app.post("/embed")
35
  def embed(req: TextRequest):
 
 
 
36
  with torch.no_grad():
37
  embedding = model.encode(
38
  req.text,
39
  convert_to_numpy=True,
40
  normalize_embeddings=True
41
  )
 
42
  return {
43
  "embedding": embedding.tolist(),
44
  "dim": len(embedding)
 
27
  class TextRequest(BaseModel):
28
  text: str
29
 
30
+ @app.get("/")
31
+ def root():
32
+ return {"message": "API is running"}
33
+
34
  @app.get("/health")
35
  def health():
36
  return {"status": "ok"}
37
 
38
  @app.post("/embed")
39
  def embed(req: TextRequest):
40
+ if model is None:
41
+ return {"error": "Model not loaded"}
42
+
43
  with torch.no_grad():
44
  embedding = model.encode(
45
  req.text,
46
  convert_to_numpy=True,
47
  normalize_embeddings=True
48
  )
49
+
50
  return {
51
  "embedding": embedding.tolist(),
52
  "dim": len(embedding)