mdanish commited on
Commit
b030067
·
verified ·
1 Parent(s): 7f65331

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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='pinterest-weightsv1-0'
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