From e656cd4e12330ea5a6a343a8da2c4dae09015dba Mon Sep 17 00:00:00 2001
From: Brian Bjarke Jensen <bbj@skov.dk>
Date: Sat, 27 Jul 2024 21:08:47 +0200
Subject: [PATCH] remove unneeded code for running in docker container

---
 model/src/main.py | 13 +++----------
 1 file changed, 3 insertions(+), 10 deletions(-)

diff --git a/model/src/main.py b/model/src/main.py
index e776beb..7069001 100644
--- a/model/src/main.py
+++ b/model/src/main.py
@@ -1,25 +1,18 @@
 """Main script to be run by service."""
 
-from pathlib import Path
-
-from dotenv import load_dotenv
 from models import VisualCommunicationModel
-from torchinfo import summary
 from utils import load_model
 
 from shared.data_store import connect
 from shared.utils import setup_logging
 
 if __name__ == '__main__':
-    # load in env file
-    env_path = Path(__file__).parent.parent.parent / 'server.env'
-    load_dotenv(env_path)
     # setup logging
     setup_logging()
     # connect to minio
     minio_client = connect()
     # instantiate model
     model: VisualCommunicationModel = load_model(client=minio_client)
-    # show model weights
-    summary(model)
-    print('loaded model')
+    # get image
+
+    # make prediction