vicliv commited on
Commit
abf6f7a
·
1 Parent(s): 1228840

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -124,7 +124,7 @@ clip_tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
124
  transform = transforms.Compose([
125
  transforms.Resize((256, 256)),
126
  transforms.ToTensor(),
127
- transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073], std=[0.26862954, 0.26130258, 0.27577711])
128
  ])
129
 
130
  def compute_prompt_match(image: Image.Image, prompt: str) -> float:
 
124
  transform = transforms.Compose([
125
  transforms.Resize((256, 256)),
126
  transforms.ToTensor(),
127
+ transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
128
  ])
129
 
130
  def compute_prompt_match(image: Image.Image, prompt: str) -> float: