Update app.py
Browse files
app.py
CHANGED
@@ -1,8 +1,5 @@
|
|
1 |
# Load model directly
|
2 |
-
|
3 |
-
import gradio as gr
|
4 |
-
#from transformers import GraphormerForGraphClassification, pipeline
|
5 |
-
from huggingface_hub import transformers
|
6 |
from transformers import GraphormerForGraphClassification
|
7 |
|
8 |
import os
|
@@ -11,5 +8,5 @@ try:
|
|
11 |
except ImportError:
|
12 |
os.system('pip install toml')
|
13 |
import toml
|
14 |
-
|
15 |
model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression")
|
|
|
1 |
# Load model directly
|
2 |
+
import transformers
|
|
|
|
|
|
|
3 |
from transformers import GraphormerForGraphClassification
|
4 |
|
5 |
import os
|
|
|
8 |
except ImportError:
|
9 |
os.system('pip install toml')
|
10 |
import toml
|
11 |
+
print('todo en orden')
|
12 |
model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression")
|