sergeipetrov commited on
Commit
118af14
1 Parent(s): 6e88679

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -7,7 +7,7 @@ import base64
7
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
10
- self.pipeline=pipeline("zero-shot-image-classification",model=path)
11
 
12
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
13
  """
 
7
 
8
  class EndpointHandler():
9
  def __init__(self, path=""):
10
+ self.pipeline=pipeline("zero-shot-image-classification",model="openai/clip-vit-large-patch14-336")
11
 
12
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
13
  """