TheMihirNaik commited on
Commit
bb18925
·
verified ·
1 Parent(s): 3944fa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
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()}