| # Load model directly | |
| from transformers import AutoTokenizer, GraphormerForGraphClassification | |
| tokenizer = AutoTokenizer.from_pretrained("PedroLancharesSanchez/graph-regression") | |
| model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression") |