Update handler.py

handler.py

@@ -6,6 +6,8 @@ from PIL import Image
 import time
 import logging
 import os
+import torch
+import torchvision
 
 
 class EndpointHandler():
@@ -18,6 +20,8 @@ class EndpointHandler():
         t1 = time.time()
         self.model = YOLO(os.path.join(path, 'yolov8m_detect_usdl.pt'))
         logging.info(f'TIME: loading the model {time.time() - t1}')
+        logging.info(f'torch version : {torch.__version__}')
+        logging.info(f'torch vision version: {torchvision.__version__}')
 
 
     def __call__(self, data: Dict) -> Dict: