mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 21:56:31 +00:00
b21f8e320f
add tensorrt engine
21 lines
828 B
Cython
21 lines
828 B
Cython
import time
|
|
|
|
cdef str CONFIG_FILE = "config.yaml" # Port for the zmq
|
|
|
|
cdef int QUEUE_MAXSIZE = 1000 # Maximum size of the command queue
|
|
cdef str COMMANDS_QUEUE = "azaion-commands"
|
|
cdef str ANNOTATIONS_QUEUE = "azaion-annotations"
|
|
|
|
cdef str API_URL = "https://api.azaion.com" # Base URL for the external API
|
|
cdef str QUEUE_CONFIG_FILENAME = "secured-config.json"
|
|
|
|
cdef str AI_ONNX_MODEL_FILE_BIG = "azaion.onnx.big"
|
|
cdef str AI_ONNX_MODEL_FILE_SMALL = "azaion.onnx.small"
|
|
|
|
cdef str AI_TENSOR_MODEL_FILE_BIG = "azaion.engine.big"
|
|
cdef str AI_TENSOR_MODEL_FILE_SMALL = "azaion.engine.small"
|
|
|
|
cdef log(str log_message, bytes client_id=None):
|
|
local_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())
|
|
client_str = '' if client_id is None else f' {client_id}'
|
|
print(f'[{local_time}{client_str}]: {log_message}') |