Chyd19 commited on
Commit
80a0056
·
verified ·
1 Parent(s): 9b6aa9b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = {