Upload app.py with huggingface_hub
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ 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='
|
31 |
)
|
32 |
tokenizer = open_clip.get_tokenizer('ViT-H-14')
|
33 |
return model, preprocess, tokenizer
|
|
|
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='dfn5b'
|
31 |
)
|
32 |
tokenizer = open_clip.get_tokenizer('ViT-H-14')
|
33 |
return model, preprocess, tokenizer
|