PK03 commited on
Commit
ee4780c
·
verified ·
1 Parent(s): 76c98b4

Update utilfuncs.py

Browse files
Files changed (1) hide show
  1. utilfuncs.py +1 -1
utilfuncs.py CHANGED
@@ -34,7 +34,7 @@ class FixedNeuronsNetwork(nn.Module):
34
  return output
35
 
36
  def load_kmeans_model(model_path):
37
- kmeans_loaded = joblib.load('kmeans_model_main.pkl')
38
  return kmeans_loaded
39
 
40
  def load_churn_model(model_path):
 
34
  return output
35
 
36
  def load_kmeans_model(model_path):
37
+ kmeans_loaded = joblib.load(model_path)
38
  return kmeans_loaded
39
 
40
  def load_churn_model(model_path):