Spaces:
Runtime error
Runtime error
update
Browse files
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((
|
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]}
|