Updating to single text input

app.py

@@ -8,11 +8,8 @@ model = SentenceTransformer("nomic-ai/nomic-embed-text-v1", trust_remote_code=Tr
 
 
 @spaces.GPU
-def embed(documents: List[str]):
-    embeddings = []
-    for document in documents:
-        embeddings.append(model.encode(document))
-    return embeddings
+def embed(document: str):
+    return model.encode(document)
 
 
 with gr.Blocks() as app: