diff --git a/dockerfiles/Dockerfile b/dockerfiles/Dockerfile
index 0709647..b7e5ff8 100644
--- a/dockerfiles/Dockerfile
+++ b/dockerfiles/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.4.1-runtime-ubuntu20.04
+FROM nvidia/cuda:11.4.3-runtime-ubuntu20.04
 
 
 RUN apt-get -y update && \
diff --git a/src/inference/inference_image.py b/src/inference/inference_image.py
index 4684ae3..0321f5f 100644
--- a/src/inference/inference_image.py
+++ b/src/inference/inference_image.py
@@ -1,4 +1,5 @@
 import os
+import cv2
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
@@ -7,7 +8,6 @@
 from torchvision import datasets,transforms,models,utils
 import numpy as np
 import matplotlib.pyplot as plt
-import os
 import pandas as pd
 from PIL import Image
 import sys