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
|