Update deployment/controller/LoadModel.py

deployment/controller/LoadModel.py

@@ -1,8 +1,8 @@
-import pickle
-
-def LoadModel(file_path: str):
-
-    with open(file_path, 'rb') as file:
-        loaded_model = pickle.load(file)
-
+import pickle
+
+def LoadModel(file_path: str):
+
+    with open(file_path, 'rb') as file:
+        loaded_model = pickle.load(file,allow_dangerous_deserialization= True)
+
     return loaded_model