Prechádzať zdrojové kódy

Changes en path of model

oscarleiva 7 rokov pred
rodič
commit
762055b405
1 zmenil súbory, kde vykonal 3 pridanie a 2 odobranie
  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)