aduzlo commited on
Commit
f1c299f
·
1 Parent(s): bd0a533

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -1
handler.py CHANGED
@@ -1,10 +1,11 @@
1
  from typing import Dict, List, Any
2
  import joblib
3
  import pandas as pd
 
4
 
5
  class EndpointHandler:
6
  def __init__(self, path=""):
7
- self.model = joblib.load("/repository/valid_trial_v3_2.joblib")
8
 
9
  def __call__(self, data: Dict[str, Any]) -> List[Any]:
10
  """
 
1
  from typing import Dict, List, Any
2
  import joblib
3
  import pandas as pd
4
+ from vtclassifier import ValidTrialClassifier
5
 
6
  class EndpointHandler:
7
  def __init__(self, path=""):
8
+ self.model = joblib.load(path + "/valid_trial_v3_2.joblib")
9
 
10
  def __call__(self, data: Dict[str, Any]) -> List[Any]:
11
  """