Spaces:

TheMihirNaik
/

generate-embeddings-api

Running

TheMihirNaik commited on Feb 28

Commit

f3e3788

verified ·

1 Parent(s): 9fe496e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,14 +8,18 @@ all_MiniLM_L6_V2_model = SentenceTransformer('sentence-transformers/all-MiniLM-L
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!"}
 @app.post("/generate-embeddings/sentence-transformers-all-MiniLM-L6-v2/")
-def generate_embeddings_model1(sentences: list, all_MiniLM_L6_V2_model):
-    sentences = [
-    "The weather is lovely today.",
-    "It's so sunny outside!",
-    "He drove to the stadium.",
-]
     embeddings = all_MiniLM_L6_V2_model.encode(sentences)
-    return {"embeddings": embeddings.tolist()}  # Convert to list for JSON serialization

 @app.get("/")
 def greet_json():
+    return f"""
+    <html>
+        <body>
+            <h1>Hello, World!</h1>
+            <a href='/generate-embeddings/sentence-transformers-all-MiniLM-L6-v2/'>Generate Embeddings</a>
+        </body>
+    </html>
+    """
+from typing import List
 @app.post("/generate-embeddings/sentence-transformers-all-MiniLM-L6-v2/")
+def generate_embeddings_all_MiniLM_L6_V2_model(sentences: List[str]):
     embeddings = all_MiniLM_L6_V2_model.encode(sentences)
+    return {"embeddings": embeddings.tolist()}  # Return embeddings as a JSON-compatible list