Update app.py

app.py

@@ -1,5 +1,14 @@
 # Load model directly
-from transformers import AutoTokenizer, GraphormerForGraphClassification
+
+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")