zjpiazza commited on
Commit
a554e4b
·
1 Parent(s): d1e849e

Add model_path to EndpointHandler

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -1,9 +1,10 @@
1
  from typing import Dict, Any
2
 
3
  class EndpointHandler:
4
- def __init__(self):
5
  """Initialize a simple handler without model loading"""
6
- pass
 
7
 
8
  def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
9
  """
 
1
  from typing import Dict, Any
2
 
3
  class EndpointHandler:
4
+ def __init__(self, model_path: str):
5
  """Initialize a simple handler without model loading"""
6
+ self.model_path = model_path
7
+ print(f"Model path: {self.model_path}")
8
 
9
  def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
10
  """