Great work, but I don't see any documentation for the inference script or even an ArgumentParser/etc. I think I can figure it out from the code, but but it would be nice for simple docking inference to use for predictions. Apologies if this information is somewhere else and I missed it.