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

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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='laion2b_s32b_b79k',
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