Update app.py
Browse files
app.py
CHANGED
@@ -29,6 +29,8 @@ Built by **Mihir Naik** 🚀
|
|
29 |
|
30 |
all_MiniLM_L6_V2_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
|
31 |
|
|
|
|
|
32 |
@app.get("/")
|
33 |
def redirect_to_docs():
|
34 |
"""
|
@@ -55,3 +57,18 @@ def generate_embeddings_all_MiniLM_L6_V2_model(sentences: List[str]):
|
|
55 |
"""
|
56 |
embeddings = all_MiniLM_L6_V2_model.encode(sentences)
|
57 |
return {"embeddings": embeddings.tolist()} # Return embeddings as a JSON-compatible list
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
|
30 |
all_MiniLM_L6_V2_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
|
31 |
|
32 |
+
intfloat_e5_large_v2_model = SentenceTransformer('intfloat/e5-large-v2')
|
33 |
+
|
34 |
@app.get("/")
|
35 |
def redirect_to_docs():
|
36 |
"""
|
|
|
57 |
"""
|
58 |
embeddings = all_MiniLM_L6_V2_model.encode(sentences)
|
59 |
return {"embeddings": embeddings.tolist()} # Return embeddings as a JSON-compatible list
|
60 |
+
|
61 |
+
|
62 |
+
@app.post("/generate-embeddings/intfloat-e5-large-v2/")
|
63 |
+
def generate_embeddings_intfloat_e5_large_v2_model(sentences: List[str]):
|
64 |
+
"""
|
65 |
+
Generates embeddings for a list of sentences using the intfloat/e5-large-v2 model.
|
66 |
+
|
67 |
+
Args:
|
68 |
+
sentences (List[str]): A list of sentences to generate embeddings for.
|
69 |
+
|
70 |
+
Returns:
|
71 |
+
dict: A dictionary containing the embeddings as a JSON-compatible list.
|
72 |
+
"""
|
73 |
+
embeddings = intfloat_e5_large_v2_model.encode(sentences)
|
74 |
+
return {"embeddings": embeddings.tolist()}
|