oshita-n commited on
Commit
e161c5d
·
1 Parent(s): c5ca716
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ def process(input_image, prompt):
7
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
8
 
9
  model, vis_processors, txt_processors = load_model_and_preprocess(name="pnp_vqa", model_type="base", is_eval=True, device=device)
10
- input_image = input_image.resize((128, 128))
11
  image = vis_processors["eval"](input_image).unsqueeze(0).to(device)
12
  text_input = txt_processors["eval"](prompt)
13
  sample = {"image": image, "text_input": [text_input]}
 
7
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
8
 
9
  model, vis_processors, txt_processors = load_model_and_preprocess(name="pnp_vqa", model_type="base", is_eval=True, device=device)
10
+ input_image = input_image.resize((256, 256))
11
  image = vis_processors["eval"](input_image).unsqueeze(0).to(device)
12
  text_input = txt_processors["eval"](prompt)
13
  sample = {"image": image, "text_input": [text_input]}