m

handler.py

@@ -1,13 +1,11 @@
 from typing import Dict, List, Any
-from comet import load_from_checkpoint
+from comet import load_from_checkpoint as lfc
 import os
 
 
 class EndpointHandler():
     def __init__(self, path=""):
-        print(os.listdir(os.getcwd()))
-
-        self.model = load_from_checkpoint("model.ckpt")
+        self.m = lfc("model.ckpt")
 
     def __call__(self, data: Dict[str, Any]) -> List[Any]:
         """
@@ -24,7 +22,7 @@ class EndpointHandler():
 
         data = inputs.pop("data")
         
-        model_output = self.model.predict(data, batch_size=batch_size, num_workers=workers, gpus=0)
+        model_output = self.m.predict(data, batch_size=batch_size, num_workers=workers, gpus=0)
         scores = model_output["scores"]
 
         return scores