antlr4-python3-runtime
torch
pillow