guidel commited on
Commit
77dfe3f
1 Parent(s): 9d9bbed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ from PIL import Image
10
  def load_clip(model_size='large'):
11
  if model_size == 'base':
12
  MODEL_name = 'openai/clip-vit-base-patch32'
13
- elif model_size == 'large':
14
- MODEL_name = 'openai/clip-vit-large-patch14'
15
 
16
  model = CLIPModel.from_pretrained(MODEL_name)
17
  processor = CLIPImageProcessor.from_pretrained(MODEL_name)
 
10
  def load_clip(model_size='large'):
11
  if model_size == 'base':
12
  MODEL_name = 'openai/clip-vit-base-patch32'
13
+ #elif model_size == 'large':
14
+ # MODEL_name = 'openai/clip-vit-large-patch14'
15
 
16
  model = CLIPModel.from_pretrained(MODEL_name)
17
  processor = CLIPImageProcessor.from_pretrained(MODEL_name)