PedroLancharesSanchez commited on
Commit
d6d77b8
1 Parent(s): 6491f44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,9 +15,9 @@ print('todo en orden')
15
  model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression")
16
 
17
  def predict(instancia):
18
- print(instancia)
19
- instancia=json.loads(instancia)
20
- instancia_preprocesada=preprocess_item(instancia)
21
  inputs={}
22
  inputs['input_nodes'] = torch.tensor([instancia_preprocesada['input_nodes']])
23
  inputs['input_edges'] = torch.tensor([instancia_preprocesada['input_edges']])
 
15
  model = GraphormerForGraphClassification.from_pretrained("PedroLancharesSanchez/graph-regression")
16
 
17
  def predict(instancia):
18
+ with open(instancia, "r") as archivo:
19
+ datos=json.load(archivo)
20
+ instancia_preprocesada=preprocess_item(datos)
21
  inputs={}
22
  inputs['input_nodes'] = torch.tensor([instancia_preprocesada['input_nodes']])
23
  inputs['input_edges'] = torch.tensor([instancia_preprocesada['input_edges']])