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()}
|