I have created a docker file while runs main.py.
The main.py takes in an argument via sys.argv[1] which is the image(file) on which inference should run on.<
sys.argv[1]