updated function name

This commit is contained in:
Brian Bjarke Jensen
2024-07-28 21:08:51 +02:00
parent 05774364fc
commit 396bf29f32
+2 -2
View File
@@ -8,14 +8,14 @@ from models import VisualCommunicationModel
from tqdm import tqdm from tqdm import tqdm
from utils import DEVICE, VCDADataset, load_model from utils import DEVICE, VCDADataset, load_model
from shared.data_store import connect from shared.data_store import connect_minio
from shared.utils import setup_logging from shared.utils import setup_logging
if __name__ == '__main__': if __name__ == '__main__':
# setup logging # setup logging
setup_logging() setup_logging()
# connect to minio # connect to minio
minio_client = connect() minio_client = connect_minio()
# instantiate model # instantiate model
model: VisualCommunicationModel = load_model(client=minio_client) model: VisualCommunicationModel = load_model(client=minio_client)
model.eval() model.eval()