2024-10-20 00:10:27 +02:00
1 changed files with 2 additions and 2 deletions
@@ -8,14 +8,14 @@ from models import VisualCommunicationModel
 from tqdm import tqdm
 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

 if __name__ == '__main__':
    # setup logging
    setup_logging()
    # connect to minio
-    minio_client = connect()
+    minio_client = connect_minio()
    # instantiate model
    model: VisualCommunicationModel = load_model(client=minio_client)
    model.eval()