LDanielBlueway commited on
Commit
dbb6faf
·
verified ·
1 Parent(s): 3888170

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -7,8 +7,8 @@ import base64
7
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
10
- self.pipeline=pipeline("zero-shot-object-detection",model="omlab/omdet-turbo-swin-tiny-hf")
11
-
12
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
13
  """
14
  data args:
 
7
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
10
+ self.pipeline=pipeline("zero-shot-object-detection",model=path)
11
+
12
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
13
  """
14
  data args: