Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ vqa_model = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-base")
|
|
| 55 |
# Metrics
|
| 56 |
clip_model, clip_preprocess = clip.load("ViT-B/32", device=device)
|
| 57 |
lpips_model = lpips.LPIPS(net='alex').to(device)
|
| 58 |
-
lpips_transform = T.Compose([T.ToTensor(), T.Resize((256,256))]
|
| 59 |
|
| 60 |
# Style presets
|
| 61 |
style_map = {
|
|
|
|
| 55 |
# Metrics
|
| 56 |
clip_model, clip_preprocess = clip.load("ViT-B/32", device=device)
|
| 57 |
lpips_model = lpips.LPIPS(net='alex').to(device)
|
| 58 |
+
lpips_transform = T.Compose([T.ToTensor(), T.Resize((256,256))])
|
| 59 |
|
| 60 |
# Style presets
|
| 61 |
style_map = {
|