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