Spaces:

oucgc1996
/

DeepACE_classification_model

Running

oucgc1996 commited on Apr 13

Commit

9c742a3

•

1 Parent(s): df23a61

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -201,7 +201,8 @@ def pdb_structure(Structure_index):
         SurfacePoitCloud_all.append(feature)
     SurfacePoitCloud_all_tensor = torch.squeeze(torch.stack(SurfacePoitCloud_all),dim=1)
     for folder in created_folders:
-        shutil.rmtree(folder)
     return SurfacePoitCloud_all_tensor
 def ACE(file):

         SurfacePoitCloud_all.append(feature)
     SurfacePoitCloud_all_tensor = torch.squeeze(torch.stack(SurfacePoitCloud_all),dim=1)
     for folder in created_folders:
+        shutil.rmtree(folder
+    print(SurfacePoitCloud_all_tensor.shape)
     return SurfacePoitCloud_all_tensor
 def ACE(file):