fix llava-next notebook gradio (#2139)

openvinotoolkit · Jun 21, 2024 · 160a9ca · 160a9ca
1 parent 3948fc3
commit 160a9ca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb b/notebooks/llava-next-multimodal-chatbot/llava-next-multimodal-chatbot.ipynb
@@ -1454,7 +1454,7 @@
     "def bot_streaming(message, history):\n",
     "    print(message)\n",
     "    if message[\"files\"]:\n",
-    "        image = message[\"files\"][-1][\"path\"]\n",
+    "        image = message[\"files\"][-1][\"path\"] if isinstance(message[\"files\"][-1], dict) else message[\"files\"][-1]\n",
     "    else:\n",
     "        # if there's no image uploaded for this turn, look for images in the past turns\n",
     "        # kept inside tuples, take the last one\n",