Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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.
|
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:
|