fix image layout for llava gradio (#2512)

openvinotoolkit · Nov 11, 2024 · 37640c3 · 37640c3
1 parent f3e0ec4
commit 37640c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/llava-multimodal-chatbot/gradio_helper.py b/notebooks/llava-multimodal-chatbot/gradio_helper.py
@@ -39,7 +39,7 @@ def read_image(path: str) -> ov.Tensor:
 
         """
         pic = Image.open(path).convert("RGB")
-        image_data = np.array(pic.getdata()).reshape(1, 3, pic.size[1], pic.size[0]).astype(np.byte)
+        image_data = np.array(pic.getdata()).reshape(1, pic.size[1], pic.size[0], 3).astype(np.byte)
         return ov.Tensor(image_data)
 
     class TextQueue: