Update app.py
Browse files
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('
|
| 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)
|