Ninjani commited on
Commit
1bc807a
·
1 Parent(s): af84992
Files changed (1) hide show
  1. app.py +1 -1
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({