Update app.py

app.py

@@ -7,8 +7,16 @@ from sentence_transformers import SentenceTransformer
 
 model = SentenceTransformer("nomic-ai/nomic-embed-text-v1.5", trust_remote_code=True)
 
-def embed(document: str):
-    '''This will embed text, normalize the embedding, and return a 768-dimension vector'''
+def embed(document: str) -> List[float]:
+    """
+    Embed text using the Nomic AI model, normalize the embedding, and return a 768-dimension vector.
+
+    Args:
+        document (str): The input text to embed.
+
+    Returns:
+        List[float]: The normalized embedding vector (length 768).
+    """
     embedding = model.encode(document)
     normalized_embedding = embedding / np.linalg.norm(embedding)
     return normalized_embedding.tolist()