ValueError when drag &drop a pic of different dimension

[jainharshit3107]

If we drag and drop a picture of different dimension other than (256 , 256) it shows "value error" of expected shape and found shape.
As I have made changes in main.py of api folder to resolve this bug, so please kindly review the PR #10