thorfine commited on
Commit
a4fdd86
·
verified ·
1 Parent(s): d951e9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ def ask_image(image, audio):
31
 
32
  # Generate the answer to the question about the image
33
  generated_ids = model.generate(**inputs)
34
- answer = processor.decode(generated_ids, skip_special_tokens=True)[0]
35
 
36
  print(f"Answer: {answer}")
37
 
 
31
 
32
  # Generate the answer to the question about the image
33
  generated_ids = model.generate(**inputs)
34
+ answer = processor.decode(generated_ids[0], skip_special_tokens=True)
35
 
36
  print(f"Answer: {answer}")
37