emiliamzz commited on
Commit
3d9aed6
·
verified ·
1 Parent(s): 2810566

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ def image_qa(image: str, question: str) -> str:
63
  question: A string representing the question being asked.
64
  """
65
  pipe = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa")
66
- answer = pipe(document, question)
67
  return answer[0]['answer']
68
 
69
  @tool
 
63
  question: A string representing the question being asked.
64
  """
65
  pipe = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa")
66
+ answer = pipe(image, question)
67
  return answer[0]['answer']
68
 
69
  @tool