logeswari commited on
Commit
2cd1c19
·
1 Parent(s): 084ac18
Files changed (1) hide show
  1. main.py +1 -3
main.py CHANGED
@@ -6,8 +6,6 @@ import numpy as np
6
  # Initialize the FastAPI app
7
  app = FastAPI()
8
 
9
- # Load the pre-trained SentenceTransformer model from Hugging Face
10
- #model = SentenceTransformer("//huggingface.co/spaces/Kabila22/Kabilan_embedding_1", trust_remote_code=True)
11
  model = SentenceTransformer("Alibaba-NLP/gte-base-en-v1.5", trust_remote_code=True)
12
 
13
  # Define the request body schema
@@ -17,7 +15,7 @@ class TextInput(BaseModel):
17
  # Home route
18
  @app.get("/")
19
  async def home():
20
- return {"message": "Welcome to embedding SMS API, use /docs to post SMS text and get dimensions"}
21
 
22
  # Define the API endpoint
23
  @app.post("/embed")
 
6
  # Initialize the FastAPI app
7
  app = FastAPI()
8
 
 
 
9
  model = SentenceTransformer("Alibaba-NLP/gte-base-en-v1.5", trust_remote_code=True)
10
 
11
  # Define the request body schema
 
15
  # Home route
16
  @app.get("/")
17
  async def home():
18
+ return {"message": "Welcome to my dashboard"}
19
 
20
  # Define the API endpoint
21
  @app.post("/embed")