Update app.py

app.py

@@ -241,7 +241,6 @@ def ACE(file):
         print("=================================Start prediction========================")
         for index, (batch, structure_fea, fingerprint) in enumerate(test_dataloader):
             batchs = {k: v for k, v in batch.items()}
-            print(structure_fea)
             outputs = model(structure_fea, batchs, fingerprint)
             probability = outputs[0].tolist()
             train_argmax = np.argmax(outputs.cpu().detach().numpy(), axis=1)