oucgc1996 commited on
Commit
1708788
·
verified ·
1 Parent(s): 7253b6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -241,6 +241,7 @@ def ACE(file):
241
  print("=================================Start prediction========================")
242
  for index, (batch, structure_fea, fingerprint) in enumerate(test_dataloader):
243
  batchs = {k: v for k, v in batch.items()}
 
244
  outputs = model(structure_fea, batchs, fingerprint)
245
  probability = outputs[0].tolist()
246
  print(outputs)
 
241
  print("=================================Start prediction========================")
242
  for index, (batch, structure_fea, fingerprint) in enumerate(test_dataloader):
243
  batchs = {k: v for k, v in batch.items()}
244
+ print(structure_fea)
245
  outputs = model(structure_fea, batchs, fingerprint)
246
  probability = outputs[0].tolist()
247
  print(outputs)