Update app.py
Browse files
app.py
CHANGED
|
@@ -202,7 +202,6 @@ def pdb_structure(Structure_index):
|
|
| 202 |
SurfacePoitCloud_all_tensor = torch.squeeze(torch.stack(SurfacePoitCloud_all),dim=1)
|
| 203 |
for folder in created_folders:
|
| 204 |
shutil.rmtree(folder)
|
| 205 |
-
print(SurfacePoitCloud_all_tensor.shape)
|
| 206 |
return SurfacePoitCloud_all_tensor
|
| 207 |
|
| 208 |
def ACE(file):
|
|
@@ -245,8 +244,6 @@ def ACE(file):
|
|
| 245 |
print(structure_fea)
|
| 246 |
outputs = model(structure_fea, batchs, fingerprint)
|
| 247 |
probability = outputs[0].tolist()
|
| 248 |
-
print(outputs)
|
| 249 |
-
print(probability)
|
| 250 |
train_argmax = np.argmax(outputs.cpu().detach().numpy(), axis=1)
|
| 251 |
for j in range(0,len(train_argmax)):
|
| 252 |
output = train_argmax[j]
|
|
|
|
| 202 |
SurfacePoitCloud_all_tensor = torch.squeeze(torch.stack(SurfacePoitCloud_all),dim=1)
|
| 203 |
for folder in created_folders:
|
| 204 |
shutil.rmtree(folder)
|
|
|
|
| 205 |
return SurfacePoitCloud_all_tensor
|
| 206 |
|
| 207 |
def ACE(file):
|
|
|
|
| 244 |
print(structure_fea)
|
| 245 |
outputs = model(structure_fea, batchs, fingerprint)
|
| 246 |
probability = outputs[0].tolist()
|
|
|
|
|
|
|
| 247 |
train_argmax = np.argmax(outputs.cpu().detach().numpy(), axis=1)
|
| 248 |
for j in range(0,len(train_argmax)):
|
| 249 |
output = train_argmax[j]
|