How do I fix a tensor dimension mismatch in TinyML disease detection? - EdgeBench