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:
|