Spaces:
Runtime error
Runtime error
fix
Browse files
app.py
CHANGED
|
@@ -11,9 +11,7 @@ def process(input_image, prompt):
|
|
| 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]}
|
| 14 |
-
|
| 15 |
-
output = model.forward_qa(samples=sample, num_captions=50)
|
| 16 |
-
pred_answers = output[0]
|
| 17 |
|
| 18 |
return pred_answer
|
| 19 |
|
|
|
|
| 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]}
|
| 14 |
+
pred_answers, caption, gradcam = model.predict_answers(sample, num_captions=50, num_patches=20)
|
|
|
|
|
|
|
| 15 |
|
| 16 |
return pred_answer
|
| 17 |
|