Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
@@ -26,9 +26,8 @@ tokenizer = open_clip.get_tokenizer(clip_model_name)
|
|
26 |
def load_model():
|
27 |
"""Load the OpenCLIP model and return model and processor"""
|
28 |
model, _, preprocess = open_clip.create_model_and_transforms(
|
29 |
-
'ViT-H-14',
|
30 |
-
pretrained='
|
31 |
-
quickgelu=True
|
32 |
)
|
33 |
tokenizer = open_clip.get_tokenizer('ViT-H-14')
|
34 |
return model, preprocess, tokenizer
|
|
|
26 |
def load_model():
|
27 |
"""Load the OpenCLIP model and return model and processor"""
|
28 |
model, _, preprocess = open_clip.create_model_and_transforms(
|
29 |
+
'ViT-H-14-378-quickgelu',
|
30 |
+
pretrained='pinterest-weightsv1-0'
|
|
|
31 |
)
|
32 |
tokenizer = open_clip.get_tokenizer('ViT-H-14')
|
33 |
return model, preprocess, tokenizer
|