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