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]
|