Commit
•
494283f
1
Parent(s):
fa7cab9
Update pipeline.py
Browse files- pipeline.py +3 -3
pipeline.py
CHANGED
@@ -12,9 +12,9 @@ class PreTrainedPipeline():
|
|
12 |
"""
|
13 |
Initialize model
|
14 |
"""
|
15 |
-
self.model = CLIPModel.from_pretrained(
|
16 |
-
self.processor = CLIPProcessor.from_pretrained(
|
17 |
-
self.tokenizer = CLIPTokenizer.from_pretrained(
|
18 |
|
19 |
def __call__(self, inputs: str):
|
20 |
"""
|
|
|
12 |
"""
|
13 |
Initialize model
|
14 |
"""
|
15 |
+
self.model = CLIPModel.from_pretrained(path)
|
16 |
+
self.processor = CLIPProcessor.from_pretrained(path)
|
17 |
+
self.tokenizer = CLIPTokenizer.from_pretrained(path)
|
18 |
|
19 |
def __call__(self, inputs: str):
|
20 |
"""
|