logeswari commited on
Commit
084ac18
·
1 Parent(s): 7b50249
Files changed (3) hide show
  1. __pycache__/main.cpython-313.pyc +0 -0
  2. main.py +1 -2
  3. requirements.txt +6 -15
__pycache__/main.cpython-313.pyc DELETED
Binary file (3.89 kB)
 
main.py CHANGED
@@ -8,8 +8,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")
12
-
13
 
14
  # Define the request body schema
15
  class TextInput(BaseModel):
 
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
14
  class TextInput(BaseModel):
requirements.txt CHANGED
@@ -1,15 +1,6 @@
1
- annotated-types==0.7.0
2
- anyio==4.8.0
3
- fastapi==0.115.8
4
- idna==3.10
5
- pydantic==2.10.6
6
- pydantic_core==2.27.2
7
- sniffio==1.3.1
8
- starlette==0.45.3
9
- typing_extensions==4.12.2
10
- sentence-transformers>=2.2.2,<3.0
11
- scikit-learn==1.3.2
12
- numpy==1.26.4
13
- pandas==2.1.4
14
- uvicorn==0.23.2
15
- huggingface_hub==0.14.1
 
1
+ fastapi
2
+ pandas
3
+ scikit-learn
4
+ joblib
5
+ uvicorn
6
+ sentence-transformers