Fixing the model hash changing when restarting the docker container

Author: hadikoub

README.md

@@ -66,7 +66,7 @@ To run the API, go the to the API's directory and run the following:
 #### Using Linux based docker:
 
 ```sh
-sudo NV_GPU=0 nvidia-docker run -itv $(pwd)/models:/models -p <docker_host_port>:4343 tensorflow_inference_api_gpu
+sudo NV_GPU=0 nvidia-docker run -itv $(pwd)/models:/models -v $(pwd)/models_hash:/models_hash -p <docker_host_port>:4343 tensorflow_inference_api_gpu
 ```
 
 The <docker_host_port> can be any unique port of your choice.

models_hash/model_hash.json

@@ -0,0 +1 @@
+{}

src/main/deep_learning_service.py

@@ -17,7 +17,7 @@ def __init__(self):
         # dictionary to hold the model instances (model_name: string -> model_instance: AbstractInferenceEngine)
         self.models_dict = {}
         # read from json file and append to dict
-        file_name = 'model_hash.json'
+        file_name = '/models_hash/model_hash.json'
         file_exists = os.path.exists(file_name)
         if file_exists:
             try:
@@ -26,7 +26,7 @@ def __init__(self):
             except:
                 self.models_hash_dict = {}
         else:
-            with open('model_hash.json', 'w'):
+            with open('/models_hash/model_hash.json', 'w'):
                 self.models_hash_dict = {}
         self.labels_hash_dict = {}
         self.base_models_dir = '/models'
@@ -62,7 +62,7 @@ def load_all_models(self):
             if key not in models:
                 del self.models_hash_dict[key]
         # append to json file
-        with open('model_hash.json', "w") as fp:
+        with open('/models_hash/model_hash.json', "w") as fp:
             json.dump(self.models_hash_dict, fp)
         return self.models_hash_dict