Spaces:
Running
Running
fix mask
Browse files
app.py
CHANGED
|
@@ -76,7 +76,7 @@ def build_weights_df(residue_predictions, chain_labels):
|
|
| 76 |
|
| 77 |
records = []
|
| 78 |
for i, seq in enumerate(seqs):
|
| 79 |
-
mask = attention_mask[i].astype(bool)
|
| 80 |
weights = pred_residues[i][mask]
|
| 81 |
for pos, w in enumerate(weights, 1):
|
| 82 |
records.append({
|
|
|
|
| 76 |
|
| 77 |
records = []
|
| 78 |
for i, seq in enumerate(seqs):
|
| 79 |
+
mask = ~(attention_mask[i].astype(bool))
|
| 80 |
weights = pred_residues[i][mask]
|
| 81 |
for pos, w in enumerate(weights, 1):
|
| 82 |
records.append({
|