File size: 421 Bytes
2d76b56
 
 
d942a9b
 
2d76b56
 
 
 
 
d942a9b
2d76b56
 
 
 
 
 
f88c21c
2d76b56
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
"""Embed input."""
import numpy
from hf_model_s_cpu import model_s
from logzero import logger

try:
    model = model_s()
except Exception as _:
    logger.exception(_)
    raise SystemExit(1) from _


def radio_embed(
    text: str,
) -> numpy.ndarray:
    """Embed input."""
    try:
        _ = model.encode(text.strip().splitlines())
    except Exception as _:
        logger.exception(_)
        raise

    return _