Update app.py
Browse files
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)
|