| # Load model directly | |
| import transformers | |
| from transformers import GraphormerForGraphClassification | |
| import os | |
| try: | |
| import toml | |
| except ImportError: | |
| os.system('pip install toml') | |
| import toml | |
| print('todo en orden') | |
| model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression") |