Gflorent
/

clip-text-feature-extraction

Feature Extraction

sentence_embedding

Model card Files Files and versions

Gflorent commited on Jul 5, 2022

Commit

494283f

·

1 Parent(s): fa7cab9

Update pipeline.py

Files changed (1) hide show

pipeline.py +3 -3

pipeline.py CHANGED Viewed

@@ -12,9 +12,9 @@ class PreTrainedPipeline():
         """
         Initialize model
         """
-        self.model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32")
-        self.processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
-        self.tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
     def __call__(self, inputs: str):
         """

         """
         Initialize model
         """
+        self.model = CLIPModel.from_pretrained(path)
+        self.processor = CLIPProcessor.from_pretrained(path)
+        self.tokenizer = CLIPTokenizer.from_pretrained(path)
     def __call__(self, inputs: str):
         """