0xnewton-superlore commited on
Commit
7a7f986
·
1 Parent(s): d26a895

change owner to superlore user account

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -16,8 +16,8 @@ class EndpointHandler():
16
  path (str, optional): Path to the pretrained model. Defaults to an empty string.
17
  image_size (int, optional): The size of the images to be processed. Defaults to 224.
18
  """
19
- self.model = CLIPModel.from_pretrained("SuperloreAI/clip-vit-large-patch14")
20
- self.processor = CLIPProcessor.from_pretrained("SuperloreAI/clip-vit-large-patch14")
21
  self.image_transform = Compose([
22
  Resize(image_size, interpolation=3),
23
  CenterCrop(image_size),
 
16
  path (str, optional): Path to the pretrained model. Defaults to an empty string.
17
  image_size (int, optional): The size of the images to be processed. Defaults to 224.
18
  """
19
+ self.model = CLIPModel.from_pretrained("Superlore/clip-vit-large-patch14")
20
+ self.processor = CLIPProcessor.from_pretrained("Superlore/clip-vit-large-patch14")
21
  self.image_transform = Compose([
22
  Resize(image_size, interpolation=3),
23
  CenterCrop(image_size),