Explorar o código

Changes en path of model

oscarleiva %!s(int64=7) %!d(string=hai) anos
pai
achega
762055b405
Modificáronse 1 ficheiros con 3 adicións e 2 borrados
  1. 3 2
      flask_api/app/server.py

+ 3 - 2
flask_api/app/server.py

@@ -34,7 +34,8 @@ def apicall():
 		raise e
 	
 	rgs = 'lbrprice_model_svr.pkl'
-	
+	dir_path = os.path.dirname(os.path.realpath(__file__))
+
 	if data.empty:
 		return(bad_request())
 	else:
@@ -42,7 +43,7 @@ def apicall():
 		print("Loading the model...")
 		loaded_model = None
 		try:
-			with open('./models/'+rgs,'rb') as f:
+			with open(dir_path+'/models/'+rgs,'rb') as f:
 				loaded_model = pickle.load(f)
 		except Exception as e:
 			print("Error loading model...", e)