Browse Source

add Exception to model load try

oscarleiva 7 years ago
parent
commit
b6cab962da
1 changed files with 7 additions and 3 deletions
  1. 7 3
      flask_api/app/server.py

+ 7 - 3
flask_api/app/server.py

@@ -41,9 +41,13 @@ def apicall():
 		#Load the saved model
 		#Load the saved model
 		print("Loading the model...")
 		print("Loading the model...")
 		loaded_model = None
 		loaded_model = None
-		with open('./models/'+rgs,'rb') as f:
-			loaded_model = pickle.load(f)
-
+		try:
+			with open('./models/'+rgs,'rb') as f:
+				loaded_model = pickle.load(f)
+		except Exception as e:
+			print("Error loading model...")
+			raise e
+			
 		print("The model has been loaded...doing predictions now...")
 		print("The model has been loaded...doing predictions now...")
 		predictions = loaded_model.predict(data)
 		predictions = loaded_model.predict(data)