zfejfiu

app.py

@@ -41,11 +41,8 @@ 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))
-    print(images[0])
-    images = Image.open(images[0][0])
-    print(images)
-    print(type(images))
+    images = [{"type": "image", "image": Image.open(image[0])} for image in images]
+    images.append({"type": "text", "text": text})
     if text == "" and not images:
         gr.Error("Please input a query and optionally image(s).")