oucgc1996 commited on
Commit
cf848bc
1 Parent(s): 7f99886

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
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]