File size: 324 Bytes
d1b6379 611a628 74a3db1 b2c5b3d 611a628 d1b6379 |
1 2 3 4 5 6 7 8 9 10 11 12 |
# Load model directly
import transformers
from transformers import GraphormerForGraphClassification
import os
try:
import toml
except ImportError:
os.system('pip install toml')
import toml
print('todo en orden')
model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression") |