fix method for converting image arr to PIL Image arr

get_llava_response.py

@@ -124,6 +124,8 @@ def get_llava_response(user_prompts: list[str],
         # image = load_image("../images/mouse.png") # previous method
         if isinstance(img, np.ndarray) and len(img.shape) == 2:
             img = Image.fromarray(img, 'L')
+        elif isinstance(img, np.ndarray):
+            img = Image.fromarray(img)
 
         image = img.convert('RGB')
         image_size = image.size