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

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-378-quickgelu',
30
- pretrained='dfn5b'
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