PedroLancharesSanchez commited on
Commit
611a628
·
verified ·
1 Parent(s): 3ea32ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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")