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