Update app.py

app.py

@@ -174,6 +174,7 @@ class MyModel(nn.Module):
         output_feature = self.dropout(self.relu(self.bn2(self.fc2(output_feature))))
         output_feature = self.dropout(self.relu(self.bn3(self.fc3(output_feature))))
         output_feature = self.dropout(self.output_layer(output_feature))
+        print(output_feature)
         return torch.softmax(output_feature,dim=1)
 
 
@@ -241,7 +242,7 @@ 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)
+            print(batchs)
             outputs = model(structure_fea, batchs, fingerprint)
             probability = outputs[0].tolist()
             print(outputs)