AdrienB134 commited on
Commit
a43ccc0
1 Parent(s): 85368ce
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def model_inference(
43
  EOS_WORDS_IDS = [id_processor.tokenizer.eos_token_id]
44
  print(type(images))
45
  print(images[0])
46
- images = Image.open(images[0][0])
47
  print(images)
48
  print(type(images))
49
  if text == "" and not images:
 
43
  EOS_WORDS_IDS = [id_processor.tokenizer.eos_token_id]
44
  print(type(images))
45
  print(images[0])
46
+ images = [Image.open(image[0][0]) for image in images]
47
  print(images)
48
  print(type(images))
49
  if text == "" and not images: