Update app.py

app.py

@@ -1,7 +1,8 @@
 # Load model directly
 
 import gradio as gr
-from huggingface_hub import AutoTokenizer, GraphormerForGraphClassification
+from huggingface_hub import transformers
+from transformers import GraphormerForGraphClassification
 
 import os 
 try:
@@ -10,5 +11,4 @@ except ImportError:
     os.system('pip install toml')
     import toml
 
-tokenizer = AutoTokenizer.from_pretrained("PedroLancharesSanchez/graph-regression")
 model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression")