Kseniia-Kholina commited on
Commit
7c8655d
·
verified ·
1 Parent(s): 299ebbf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def process_sequence(sequence, domain_bounds, n):
62
 
63
  plt.figure(figsize=(15, 8))
64
  sns.heatmap(transposed_logits_array, cmap='plasma', xticklabels=x_tick_labels, yticklabels=filtered_tokens)
65
- plt.title('Logits for masked per residue tokens')
66
  plt.ylabel('Token')
67
  plt.xlabel('Residue Index')
68
  plt.yticks(rotation=0)
 
62
 
63
  plt.figure(figsize=(15, 8))
64
  sns.heatmap(transposed_logits_array, cmap='plasma', xticklabels=x_tick_labels, yticklabels=filtered_tokens)
65
+ plt.title('Token Probability Heatmap')
66
  plt.ylabel('Token')
67
  plt.xlabel('Residue Index')
68
  plt.yticks(rotation=0)