| # Load model directly | |
| import gradio as gr | |
| from huggingface_hub import AutoTokenizer, GraphormerForGraphClassification | |
| import os | |
| try: | |
| import toml | |
| except ImportError: | |
| os.system('pip install toml') | |
| import toml | |
| tokenizer = AutoTokenizer.from_pretrained("PedroLancharesSanchez/graph-regression") | |
| model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression") |