19 lines
445 B
Python
19 lines
445 B
Python
"""Main script to be run by service."""
|
|
|
|
from models import VisualCommunicationModel
|
|
from utils import load_model
|
|
|
|
from shared.data_store import connect
|
|
from shared.utils import setup_logging
|
|
|
|
if __name__ == '__main__':
|
|
# setup logging
|
|
setup_logging()
|
|
# connect to minio
|
|
minio_client = connect()
|
|
# instantiate model
|
|
model: VisualCommunicationModel = load_model(client=minio_client)
|
|
# get image
|
|
|
|
# make prediction
|