Qi

First of all, thank you very much for your very good work！！！ I tried to add an **inference()** function to **main.py** to check the inference effect. But the result is...