sophiaaez commited on
Commit
85ef56a
1 Parent(s): 1612f1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -15
app.py CHANGED
@@ -29,21 +29,6 @@ model.eval()
29
  model = model.to(device)
30
 
31
 
32
- from models.blip_vqa import blip_vqa
33
-
34
- image_size_vq = 480
35
- transform_vq = transforms.Compose([
36
- transforms.Resize((image_size_vq,image_size_vq),interpolation=InterpolationMode.BICUBIC),
37
- transforms.ToTensor(),
38
- transforms.Normalize((0.48145466, 0.4578275, 0.40821073), (0.26862954, 0.26130258, 0.27577711))
39
- ])
40
-
41
- model_url_vq = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model*_vqa.pth'
42
-
43
- model_vq = blip_vqa(pretrained=model_url_vq, image_size=480, vit='base')
44
- model_vq.eval()
45
- model_vq = model_vq.to(device)
46
-
47
  def getModelPath(language):
48
  if language == 'English':
49
  path = None
 
29
  model = model.to(device)
30
 
31
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
32
  def getModelPath(language):
33
  if language == 'English':
34
  path = None