File size: 405 Bytes
d1b6379 b2c5b3d d1b6379 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
# 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") |