zvrev

app.py

@@ -42,7 +42,9 @@ def model_inference(
     BAD_WORDS_IDS = id_processor.tokenizer(["<image>", "<fake_token_around_image>"], add_special_tokens=False).input_ids
     EOS_WORDS_IDS = [id_processor.tokenizer.eos_token_id]
     print(type(images))
-    images = images[0]
+    print(images[0])
+    images = images[0][0]
+    print(images)
     print(type(images))
     if text == "" and not images:
         gr.Error("Please input a query and optionally image(s).")