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