Spaces:

oucgc1996
/

DeepACE_classification_model

Running

oucgc1996 commited on Apr 13

Commit

c6d5045

•

1 Parent(s): 9c742a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -201,7 +201,7 @@ 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
     print(SurfacePoitCloud_all_tensor.shape)
     return SurfacePoitCloud_all_tensor

         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