# 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")