rbanfield commited on
Commit
ef53d28
1 Parent(s): 22ad839

Change from rbanfield to openai vit model

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -13,8 +13,8 @@ logger.setLevel('INFO')
13
 
14
  class EndpointHandler():
15
  def __init__(self, path=""):
16
- self.model = CLIPModel.from_pretrained("rbanfield/clip-vit-large-patch14").to(device)
17
- self.processor = CLIPProcessor.from_pretrained("rbanfield/clip-vit-large-patch14")
18
 
19
  def __call__(self, data):
20
  try:
 
13
 
14
  class EndpointHandler():
15
  def __init__(self, path=""):
16
+ self.model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14").to(device)
17
+ self.processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
18
 
19
  def __call__(self, data):
20
  try: