Spaces:
Running
on
T4
Running
on
T4
File size: 381 Bytes
1668c89 3cded50 |
1 2 3 4 5 6 7 8 9 10 |
from typing import List
from infinity_emb import AsyncEmbeddingEngine
async def embed_text(engine: AsyncEmbeddingEngine, sentences: List[str]):
async with engine:
embeddings, usage = await engine.embed(sentences=sentences)
await engine.astart()
embeddings, usage = await engine.embed(sentences=sentences)
await engine.astop()
return embeddings, usage |