Spaces:
Runtime error
Runtime error
Commit
·
b13ea5a
1
Parent(s):
706fcd8
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def update(inp, file, mode, custom_resids, clustering_threshold):
|
|
| 58 |
else:
|
| 59 |
ids = get_all_metalbinding_resids(filepath)
|
| 60 |
|
| 61 |
-
voxels, prot_centers, prot_N, prots = processStructures(
|
| 62 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 63 |
voxels.to(device)
|
| 64 |
print(voxels.shape)
|
|
|
|
| 58 |
else:
|
| 59 |
ids = get_all_metalbinding_resids(filepath)
|
| 60 |
|
| 61 |
+
voxels, prot_centers, prot_N, prots = processStructures(filepath, ids)
|
| 62 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 63 |
voxels.to(device)
|
| 64 |
print(voxels.shape)
|