Issue running vim3 c++ demo with one class Yolo

Which system do you use? Android, Ubuntu, OOWOW or others?

Ubuntu

Which version of system do you use? Please provide the version of the system here:

Ubuntu 24.04.2 LTS

Please describe your issue below:

I have successfully converted the yolov8n model to .nb and was able to run it with this result:

I changed the number of classes inside of post_process function to only 1 to make it run. However, the results were terrible. After retraining my model once again on the same data but with 79 empty classes and focusing on only one that had data the model now works as intended. But is there a way to make it run without adding these empty classes? I think the issue could be in Post Process but can't really see where. Please help... :(