Added dimensionality check to avoid bad array dimensions (#5915)
* Added dimensionality check to avoid bad array dimensions
Added a simole dimensionality check to keep the code coherent
* lint
* add changeset
---------
Co-authored-by: Abubakar Abid <abubakar@huggingface.co>
Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>
